python - 在 DataFrame 中查找质心时出现 "ValueError: cannot evaluate a numeric op with unequal lengths"

Question

为什么以下失败？

import pandas as pd
import numpy as np
df = pd.DataFrame({1:[1,2,3,], 2:[4,5,6]},index=np.linspace(0,1,3))
df.T*df.index


ValueError: cannot evaluate a numeric op with unequal lengths

这不应该给我一条记录，其中每个列元素都乘以数据帧的索引吗？

奇怪的是，这有时在我的代码中有效。它似乎取决于输入的 DataFrame。

就上下文而言，我想这样做的原因是我有一个 DataFrame，其中索引是位置，元素是权重。我想使用这些权重和位置获取每列的质心。为此，我发现以下公式有效(结果看似合理)，但有时 DataFrame 会破坏它，从而导致上述错误

centroids = (df_mat.T*df_mat.index).T.sum()/df_mat.sum()

Answer 1

这并没有回答我最初的“为什么会发生错误”的问题，但是用 df.index.values` 替换 df.index 可以解决问题:

import pandas as pd
import numpy as np
df = pd.DataFrame({1:[1,2,3,], 2:[4,5,6]},index=np.linspace(0,1,3))
new_df = df.T*df.index.values
print new_df.T

达到预期效果:

     1    2
0.0  0  0.0
0.5  1  2.5
1.0  3  6.0