熊猫的加权平均值-字符串索引必须为整数

Question

我将基于csv文件计算加权平均值。我已经加载了列：A，B，其中包含浮点值。 我的csv文件：

A   B
170.804 2854
140.924 510
164.842 3355

模式

(w1*x1 + w2*x2 + ...) / (w1 + w2 + w3 + ...)

我的代码：

c = df['B']                                     # ok
wa = (df['B'] * df['A']).sum() / df['B'].sum()  # TypeError: string indices must be integers

Answer 1

IIUC，您可以尝试一下（您编写的代码行也应该可以）：

wa = df['A'].dot(df['B']) / df['B'].sum()

print(wa)

165.55897693109094