Python Pandas中如何高效地根据上一行值创建并累加新列？

wufei123 2025-03-12 阅读:17 评论:0

本文探讨如何在Python Pandas中，根据上一行值高效创建并累加新列。问题并非简单地使用apply函数就能解决，需要结合Pandas特性巧妙实现。假设数据包含三列('col1', 'col2', 'col3')，目标是根据'c...

python pandas中如何高效地根据上一行值创建并累加新列？

本文探讨如何在Python Pandas中，根据上一行值高效创建并累加新列。问题并非简单地使用apply函数就能解决，需要结合Pandas特性巧妙实现。

假设数据包含三列('col1', 'col2', 'col3')，目标是根据'col1'创建'col4'列：'col1'值在(3, 5)之间则'col4'为1，否则为0；且'col4'中值为1的连续数值进行累加。

直接使用循环遍历方法虽然清晰，但效率低：

PHP

values = [[5.5, 2.5, 10.0], [2.0, 4.5, 1.0], [2.5, 5.2, 8.0], [4.5, 5.8, 4.8], [4.6, 6.3, 9.6], [4.1, 6.4, 9.0],
          [5.1, 2.3, 11.1], [4.8, 2.3, 4.8], [4.8, 2.3, 4.8]]

df = pd.DataFrame(values, columns=['col1', 'col2', 'col3'], index=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'y'])
col4 = []
index = 1
for one in df['col1']:
    if 3 < one < 5:
        index += 1
        col4.append(index)
    else:
        index = 0
        col4.append(index)
df['col4'] = col4

自定义函数结合apply方法虽然有所改进，但不够简洁高效。

更高效的方案是利用numpy.where和cumsum函数：

PHP

df['col4'] = pd.Series(np.where((df['col1'] > 3) & (df['col1'] < 5), 1, np.nan)).cumsum().fillna(0).astype(int)

此方法先根据条件创建包含1和NaN的Series，再用cumsum累加，最后用fillna将NaN替换为0。

更稳健的版本，能处理各种情况，包括中间出现不符合条件的情况：

PHP

v = df.apply(lambda x: 1 if 3 < x['col1'] < 5 else 0, axis=1)
df['col4'] = (v.cumsum() - (v != v.shift()).cumsum() + (v != v.shift()) * v).fillna(0).astype(int)