Python/Pandas 条件多列划分(示例代码)

Question

有一个像这样的 DataFrame:

type  A  B  total
1     4  5   9
2     5  5  10
1     4  1  5

是否可以应用此代码:

df[['A','B']] = df[['A','B']].div(df['total'], axis=0)

但是基于条件，我的意思是仅当类型 == 1 时才应用它，同时保持其他值(类型 == 2)不变并且不修改(子集化)数据框作为除法的结果？

Answer 1

您可以使用 bool 掩码；事实上，pd.DataFrame.mask在这里工作:

cond = df['type'].eq(1)

df[['A', 'B']] = df[['A', 'B']].mask(cond, df[['A','B']].div(df['total'], 0))

print(df)

   type         A         B  total
0     1  0.444444  0.555556      9
1     2  5.000000  5.000000     10
2     1  0.800000  0.200000      5