索引错误:布尔索引与维度 0 上的索引数组不匹配

Question

我的代码工作正常，直到我将 Numpy 更新到 1.13.1。现在我收到以下错误

IndexError: boolean index did not match indexed array along dimension 0; dimension is 5 but corresponding boolean dimension is 4

。扔到这一行：

m = arr[np.diff(np.cumsum(arr) >= sum(arr) * i)]

我似乎无法理解它。有什么建议吗？

这是我的示例代码：

a = [1,2,3,4,5]
l = [0.85,0.90]
s = sorted(a, reverse = False)
arr = np.array(s)
for i in l:
m = arr[np.diff(np.cumsum(arr) >= sum(arr) * i)]

Answer 1

np.diff是一个比data_array小的元素。

输出的形状与 相同，但沿轴 尺寸小 n 个。

numpy.diff

我使用的是 Numpy 1.11，而不是IndexError我得到了一个VisibleDeprecationWarning.所以我想不再容忍使用不正确的尺寸。

您需要定义所需的行为，例如从第二个元素开始，或删除最后一个元素：

arr = np.array([1,2,3,4,5])
arr2 = arr[:-1]
m = arr2[np.diff(np.cumsum(arr) >= sum(arr))]
arr3 = arr[1:]
m = arr3[np.diff(np.cumsum(arr) >= sum(arr))]