获取列表类型数据字段的第一个元素

时间:2019-03-01 17:27:25

标签: python pandas

我有一个带有包含列表的单元格的数据框。 现在,我想获取列表中的每个第一个元素(如果有)。 像

416                      []
417                      []
419                   [562]
425              [570, 632]
426                      []
427    [511, 586, 753, 604]
428                      []

我想要的是类似这样的东西(可以用来创建新行)

416                    nan
417                    nan
419                    562
425                    570
426                    nan
427                    511
428                    nan

我该怎么做?还是这种不良做法,应该避免?

1 个答案:

答案 0 :(得分:2)

使用:

df['col'].str[0]

示例

df = pd.DataFrame({'col': {0: [],
  1: [],
  2: [562],
  3: [570, 632],
  4: [],
  5: [511, 586, 753, 604],
  6: []}})

print(df['col'].str[0])

0      NaN
1      NaN
2    562.0
3    570.0
4      NaN
5    511.0
6      NaN