Question

给出以下数据框：

import pandas as pd
df=pd.DataFrame({'A':['a','b','c'],
                 'B':[[[1,2],[3,4],[5,6]],[[1,2],[3,4],[5,6]],[[1,2],[3,4],[5,6]]]})
df

    A               B
0   a   [[1, 2], [3, 4], [5, 6]]
1   b   [[1, 2], [3, 4], [5, 6]]
2   c   [[1, 2], [3, 4], [5, 6]]

我想创建一个新列（＆＃39; C＆＃39;），其中包含B列元组的每个元素中的第一个值，如下所示：

    A               B                 C
0   a   [[1, 2], [3, 4], [5, 6]]   [1,3,5]
1   b   [[1, 2], [3, 4], [5, 6]]   [1,3,5]
2   c   [[1, 2], [3, 4], [5, 6]]   [1,3,5]

到目前为止，我已经尝试过了：

df['C']=df['B'][0]

...但只返回第一个元组（[1,2]）。

提前致谢！

Answer 1

df['C'] = df['B'].apply(lambda x: [y[0] for y in x])

Answer 2

试试这个：

df['C'] =   df["B"].apply(lambda x : [y[0] for y in list(x)])

Answer 3

这对我有用 -

df['C'] = df['B'].str[0]

熊猫获得细胞中每个元组的第一个元素

3 个答案: