python - 在 pandas 列表列中添加特定列表元素

Question

我有一个 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])

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

Answer 1

使用DataFrame.iloc :

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