python - 输入要列出的列名列表，然后按数据框中的特定列计算

Question

我有一个包含特定列的宽数据集，我想乘以具有总体权重的另一列，并在完成后替换这些值。当我在下面运行（示例）代码时，我得到错误：AttributeError：'function'对象没有属性'list'。

请告知我如何使用 list完成这项工作。谢谢！

df = pd.DataFrame(np.random.randint(0,100,size=(15, 4)), columns=list('ABCD'))
df['WGT']=0.5
cols_to_calc=['A', 'C']
df.update(df.columns.isin.list(cols_to_calc)).mul(df['WGT'])

标签： pythonpandasdataframe

Answer 1

做

df[cols_to_calc] = df[cols_to_calc].mul(df['WGT'], axis=0)