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

Question

我有一个 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”方式吗？

Answer 1

一如既往，请记住在帧中存储非标量对象通常是不受欢迎的，实际上应该只用作临时的中间步骤。

也就是说，您可以使用 .str 访问器，即使它不是一列字符串:

>>> df = pd.DataFrame({"A": [[1,2],[3,4],[8,9],[2,6]]})
>>> df["new_col"] = df["A"].str[0]
>>> df
        A  new_col
0  [1, 2]        1
1  [3, 4]        3
2  [8, 9]        8
3  [2, 6]        2
>>> df["new_col"]
0    1
1    3
2    8
3    2
Name: new_col, dtype: int64