Pandas :将权重应用于另一列，仅针对特定ids

import pandas as pd

# 创建一个示例DataFrame
data = {'id': [1, 2, 3, 4, 5],
        'value': [10, 20, 30, 40, 50],
        'weight': [0.1, 0.2, 0.3, 0.4, 0.5]}
df = pd.DataFrame(data)

# 定义一个函数来计算权重应用于另一列的结果
def apply_weight(row):
    if row['id'] in [2, 4]:  # 只针对特定ids进行权重应用
        return row['value'] * row['weight']
    else:
        return row['value']

# 使用apply函数将权重应用于另一列
df['result'] = df.apply(apply_weight, axis=1)

# 打印结果
print(df)

   id  value  weight  result
0   1     10     0.1    10.0
1   2     20     0.2     4.0
2   3     30     0.3    30.0
3   4     40     0.4    16.0
4   5     50     0.5    50.0