python - 如何使用Python数据框基于多个条件进行计算？

Question

我有包含数千行和列的 Excel 数据文件。我正在使用 python 并开始使用 pandas dataframes 来分析数据。我想要在 D 列中执行的操作是计算每个 ID 每年 C 列中值的年度变化。我可以使用 Excel 来执行此操作 - 如果组织 ID 与前一行中的组织 ID 相同，则计算年度变化(使单元格以蓝色突出显示，因为这是该特定 ID 的第一个期间)。我不知道如何使用 python 来做到这一点。有人可以帮忙吗？

Answer 1

假设数据帧已经排序

df.groupby(‘ID’).Cash.pct_change()

但是，假设事情已排序，您可以加快速度。因为不需要分组来计算从一行到下一行的百分比变化

df.Cash.pct_change().mask(
    df.ID != df.ID.shift()
)

这些应该会生成您正在查找的列值。为了添加列，您需要分配给列或使用新列创建一个新数据框

df[‘AnnChange’] = df.groupby(‘ID’).Cash.pct_change()