在Python中,可以使用pandas
库来按照一个因素列进行分组并对结果进行加权。以下是一个示例代码:
import pandas as pd
# 创建一个包含因素列和权重列的DataFrame
data = {'因素列': ['A', 'B', 'A', 'B', 'B', 'A'],
'权重列': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
# 按照因素列进行分组,并根据权重列进行加权求和
grouped = df.groupby('因素列').apply(lambda x: sum(x['权重列'] * x['权重列']))
print(grouped)
输出结果为:
因素列
A 35
B 65
dtype: int64
在上述代码中,首先创建了一个包含因素列和权重列的DataFrame。然后,使用groupby
方法按照因素列进行分组。接下来,通过apply
方法对每个分组进行操作,将权重列与权重列相乘,并求和。最后,打印出分组结果。