在数据帧上循环并使用 if 语句

Question

我运行在遍历 pandas 数据框中的行时遇到了问题

这是我正在尝试的代码运行

data = {'test':[1,1,0,0,3,1,0,3,0],
                'test2':[0, 2, 0,1,1,2,7,3,2],
                }
df = pd.DataFrame(data)
df['combined'] = df['test'] +df['test2']
df['combined'].astype('float64')
df
    
for index, row in df.iterrows():
    if row['test']>=1 & row['test2']>=1:
        row['combined']/=2
    else:
        pass

所以，如果 test 和 test2 的值都为 1 或更大，它应该除以 2，但是它不会除以所有应该除以的行。

我是不是哪里弄错了？

这是我运行代码时的结果相应的列是 test、test2 和 combined

0   1   0   1
1   1   2   3
2   0   0   0
3   0   1   1
4   3   1   2
5   1   2   3
6   0   7   7
7   3   3   3
8   0   2   2

Answer 1

你所做的通常是一种不好的做法，因为如果不是绝对必要的话，出于性能原因应该避免迭代行，解决方案是根据你的条件定义掩码并使用 .loc 在掩码内操作：

data = {'test':[1,1,0,0,3,1,0,3,0],
                'test2':[0, 2, 0,1,1,2,7,3,2],
                }
df = pd.DataFrame(data)
df['combined'] = df['test'] +df['test2']
df['combined'].astype('float64')
mask = (df['test']>=1) & (df['test2']>=1)
df.loc[mask,'combined'] /=2

Answer 2

您正在使用 &，按位与运算符。您应该使用 and，布尔 AND 运算符。这导致 if 语句给出您不期望的答案。

在数据帧上循环并使用 if 语句

looping and with if statement over dataframe

python

loops

if-statement