访问嵌套列表中的第一个和最后一个元素

Question

我的数据帧包含嵌套列表。我想保留子列表中的第一个和最后一个元素，然后平展列表。e、 g.：

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的新手，所以我对双环的挑战有点过头了

Answer 1

用这个行：-你知道吗

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]

Answer 2

您甚至可以在创建数据帧后执行此操作：

df['names'] = df.names.apply(lambda x: [ ' '.join([i[0],i[-1]]) for i in x] )