Доступ к первым и последним элементам во вложенных списках - PullRequest
0 голосов
/ 29 июня 2019

Мой DataFrame содержит вложенные списки.Я хочу сохранить первый и последний элементы в подсписках, а затем расклеить списки.Например:

df1=pd.DataFrame({'names':[[['harry','james','potter'],['Albus','Dumbledore']],[['Peter','D.','Griffin'],['some','nice','string']]]})
df

и результат должен выглядеть следующим образом

0  ['Harry Potter','Albus Dumbledore']
1  ['Peter Griffin','some string']

Поскольку я новичок в Paython, я немного перегружен двойной петлей

Ответы [ 2 ]

3 голосов
/ 29 июня 2019

Вы даже можете сделать это после создания фрейма данных:

df['names'] = df.names.apply(lambda x: [ ' '.join([i[0],i[-1]]) for i in x] )
2 голосов
/ 29 июня 2019

Используйте эту строку: -

df1["names"] = [[j[0] + " " + j[-1]  for j in i] for i in df1["names"]]

Выход

                              names
0  [harry potter, Albus Dumbledore]
1      [Peter Griffin, some string]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...