Python中np.histogram报ValueError：weights与a形状不匹配的解决求助

问题出在长度一致≠形状一致，len()只能告诉你第一维度的元素个数，但numpy判断数组匹配是看完整的shape属性。比如你的costh可能是二维数组（比如形状是(10000, 1)），而weight是一维数组（形状(10000,)），这时候两者形状就不匹配，即使第一维度长度都是10000。

解决步骤： #

1. 先检查两者的完整形状：

import numpy as np
# 打印完整形状，别只看长度
print(costh.shape, weight.shape)

2. 如果形状不匹配，把其中一个展平成一维数组：

   • 若costh是二维，展平它：

   costh = costh.flatten()
   

   • 若weight是二维，展平权重：

   weight = weight.flatten()
   

   也可以用np.ravel()，它和flatten()的区别是ravel()尽量返回原数组的视图（更省内存），而flatten()总是返回副本。

3. 重新执行直方图代码：

passw = np.histogram(costh, bins=10, range=(-1.0, 1.0), weights=weight)

补充说明： #

如果你的costh或weight原本是Python列表，比如costh是嵌套列表（比如[[0.1], [0.2], ...]），numpy会把它转换成二维数组，而普通列表[0.1, 0.2, ...]会变成一维数组，这时候就会出现形状不匹配的问题，展平后就能解决。

内容的提问来源于stack exchange，提问作者si wang