Pandas 如何防止自己遍历行？

Question

我有以下 dataframe

Charge_type Amount Quantity
Credit        2.5    3 
Credit        3.24   2
Debit         5.98   6

我需要以下 output，如果 Charge_type 是“信用”，则将“金额”* -1 和“数量”* -1 相乘以获得以下结果：

Charge_type Amount Quantity
Credit       -2.5    -3 
Credit       -3.24   -2
Debit         5.98    6

我不知道从哪里开始，我在网上看到的所有地方都说不要在 pandas 中迭代。 抱歉，我仍然是 Python 和 Pandas 的初学者。 任何帮助表示赞赏。

Answer 1

一种选择是使用loc ：

df.loc[df['Charge_type'].eq('Credit'), ['Amount', 'Quantity']] *= -1

第一部分创建 Boolean 索引df['Charge_type'].eq('Credit') （根据 Charge_type 为 'Credit' 的位置到 select 行）。

第二部分选择影响['Amount', 'Quantity']的列，在这种情况下，'Amount'和'Quantity'。

最后，使用快捷运算符*=进行变异以否定值。

df ：

  Charge_type  Amount  Quantity
0      Credit   -2.50        -3
1      Credit   -3.24        -2
2       Debit    5.98         6

Answer 2

您可以为Charge_type列创建掩码，并使用.loc和列名来访问掩码具有True值的列，然后相应地分配新值：

mask = df['Charge_type'] == 'Credit'
df.loc[mask, 'Amount'] = -df['Amount']
df.loc[mask, 'Quantity'] = -df['Quantity']

OUTPUT ：

  Charge_type  Amount  Quantity
0      Credit   -2.50        -3
1      Credit   -3.24        -2
2       Debit    5.98         6

Answer 3

我认为结合其他两个答案的方法（已经赞成）是最易读的：

mask = df['Charge_type'] == 'Credit'
df.loc[mask, ['Amount', 'Quantity']] *= -1

Output：

  Charge_type  Amount  Quantity
0      Credit   -2.50        -3
1      Credit   -3.24        -2
2       Debit    5.98         6