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

Question

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

标签： pythonpandaslist

Answer 1

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

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