在列表的熊猫列中添加特定的列表元素[重复]

Posted 2023-02-23

【中文标题】在列表的熊猫列中添加特定的列表元素[重复]

【英文标题】：Adding specific list-elements in a pandas column of lists [duplicate]

【发布时间】：2020-03-13 16:29:14

【问题描述】：

我有一个 pandas 数据框，其中一列由列表组成：

import pandas as pd
df = pd.DataFrame("a" : [[1,2,3], [4,5,6], [7,8,9]])

我想添加，例如，每个列表的第一个和最后一个索引。为此我使用apply:

df['a'].apply(lambda x: x[0] + x[-1])

这单独作用于每一行，如果数据框很大，这可能会非常密集。有没有办法向量化这个操作？

【问题讨论】：

使用df.iloc[:, 0]+df.iloc[:, -1]

@jezrael 请参阅 OP。抱歉，我在提出问题时（又犯了一个错误！）。现在应该是正确的。

【参考方案1】：

使用DataFrame.iloc:

df.iloc[0]+df.iloc[-1]
a     4
b    10
c    16
dtype: int64

【讨论】：

谢谢。我在我的 OP 中非常不清楚，对此我深表歉意 - 我需要 axis=1（请参阅更新后的帖子）