如何从pandas Dataframe中包含数组的单元格中提取值？

Question

以下是熊猫数据帧格式的数据源

    cinema       kiwi    restaurant         sushi
0  [2, 2, 2]  [5, 5, 5]     [2, 2, 2]  [14, 14, 14]
1  [2, 2, 2]  [2, 2, 2]     [2, 2, 2]     [2, 2, 2]
2  [1, 1, 1]  [1, 1, 1]  [10, 10, 10]     [1, 1, 1]

我想通过只选择数组的第一个值来转换它

^{pr2}$

有什么功能可以做到这一点吗？ 我不确定我用的是最直接的方法。在

提前感谢您的意见。在

Answer 1

最直接的方法是使用.applymap：

In [3]: df
Out[3]:
      cinema       kiwi    restaurant         sushi
0  [2, 2, 2]  [5, 5, 5]     [2, 2, 2]  [14, 14, 14]
1  [2, 2, 2]  [2, 2, 2]     [2, 2, 2]     [2, 2, 2]
2  [1, 1, 1]  [1, 1, 1]  [10, 10, 10]     [1, 1, 1]

In [4]: df.applymap(lambda x: x[0])
Out[4]:
   cinema  kiwi  restaurant  sushi
0       2     5           2     14
1       2     2           2      2
2       1     1          10      1