python - Pandas - 使用 .apply() 根据条件更新行中的值

Question

简化，我想使用一个函数来检查一行的条件并相应地设置值:

def helper(row):
    if row["A"] == "TEST":
         row["B"] = "WOW"
     else:
         row["C"] = "NO_GO"


moddf = df.apply(helper, axis=1)

我可以使用 iterrows() 执行此操作，但 .apply 应该更快地迭代 df 中超过 100 万行。

Answer 1

您不需要(也不应该使用)申请:

# toy data
df = pd.DataFrame({'A':['TEST','NO'],
              'B' : ['A','B'],
              'C' :list('12')})

s = df['A']=='TEST'

df.loc[s,'B'] = 'WOW'
df.loc[~s, 'C'] = 'NO_GO'

输出:

      A    B      C
0  TEST  WOW      1
1    NO    B  NO_GO