Accessing every 1st element of Pandas DataFrame column containing lists

后端 未结 4 1391
青春惊慌失措
青春惊慌失措 2020-11-27 06:56

I have a Pandas DataFrame with a column containing lists objects

      A
0   [1,2]
1   [3,4]
2   [8,9] 
3   [2,6]

How can I access the firs

4条回答
  •  猫巷女王i
    2020-11-27 07:36

    As always, remember that storing non-scalar objects in frames is generally disfavoured, and should really only be used as a temporary intermediate step.

    That said, you can use the .str accessor even though it's not a column of strings:

    >>> 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
    

提交回复
热议问题