在 pandas 列中提取列表的元素

Question

我有一个 DataFrame，其中每列都包含一个列表，如下例所示，只有两列。

     Gamma   Beta
0   [1.4652917656926299, 0.9326935235505321, float] [91, 48.611034768515864, int]
1   [2.6008354611105995, 0.7608529935313189, float] [59, 42.38646954167245, int]
2   [2.6386970166722348, 0.9785848171888037, float] [89, 37.9011122659478, int]
3   [3.49336632573625, 1.0411524946972244, float]   [115, 36.211134224288344, int]
4   [2.193991200007534, 0.7955134305428825, float]  [128, 50.03563864975485, int]
5   [3.4574527664490997, 0.9399880977511021, float] [120, 41.841146628802875, int]
6   [3.1190582380554863, 1.0839109431114795, float] [148, 55.990072419824514, int]
7   [2.7757359940789916, 0.8889801332053203, float] [142, 51.08885697101243, int]
8   [3.23820908493237, 1.0587479742892683, float]   [183, 43.831293356668425, int]
9   [2.2509032790941985, 0.8896196407231622, float] [66, 35.9377662201882, int]

我想为每一列提取每行列表的第一个位置，以获得如下所示的 DataFrame。

     Gamma   Beta
0   1.4652917656926299  91
1   2.6008354611105995  59
2   2.6386970166722348  89
...

到目前为止，我的解决方案类似于 [row[1][0] for row in df_params.itertuples()] ，我可以对行的每个列索引进行迭代，然后组成我的新 DataFrame。

另一种方法是 new_df = df_params['Gamma'].apply(lambda x: x[0]) 然后迭代遍历所有列。

我的问题是，是否有更简单的方法来执行此操作？

原文由 Ignacio Vergara Kausel 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

您可以使用 str 列表访问器，例如：

 df_params['Gamma'].str[0]

这应该适用于所有列：

 df_params.apply(lambda col: col.str[0])

原文由 IanS 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

Itertuples 会很慢。您可以通过以下方式加快速度：

 for column_name in df_params.columns:
    df_params[column_name] = [i[0] for i in df_params[column_name]]

原文由 A.Kot 发布，翻译遵循 CC BY-SA 3.0 许可协议