【发布时间】: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。抱歉,我在提出问题时(又犯了一个错误!)。现在应该是正确的。