有条件地在多列上设置值

Question

我有一组将组件反映到另一列的列。 当该列为负数时，我需要将其归零。 我还需要将所有组件列清零。 我该怎么做呢？

filter_col = [col for col in t18_df if col.startswith('aum_')]
t18_df.loc[t18_df['AUM'] < 0] # these rows should have zeroed values only on filter_col

Answer 1

您可以将filter_col传递给loc ：

filter_col = [col for col in t18_df if col.startswith('aum_')]

# zero out filter_col
t18_df.loc[t18_df['AUM'] < 0, filter_col] = 0

# zero out AUM itself
t18_df.loc[t18_df['AUM'] < 0, 'AUM'] = 0
# also
# t18_df['AUM'] = t18_df['AUM'].clip(0)