在列表的熊猫列中添加特定的列表元素

Question

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

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