访问包含列表的Pandas DataFrame列的每个第一个元素

一尘不染

python

我有一个带有包含列表对象的列的Pandas DataFrame

      A
0   [1,2]
1   [3,4]
2   [8,9] 
3   [2,6]

如何访问每个列表的第一个元素并将其保存到DataFrame的新列中？要获得这样的结果：

      A     new_col
0   [1,2]      1
1   [3,4]      3
2   [8,9]      8
3   [2,6]      2

我知道这可以通过遍历每一行来完成，但是有什么“ pythonic”方法吗？

阅读 145

2020-12-20

共1个答案

一尘不染

您可以使用map和lambda功能

df.loc[:, 'new_col'] = df.A.map(lambda x: x[0])

2020-12-20