IndexError:boolean index与维度0的索引数组不匹配

时间:2017-07-20 07:07:33

标签: python python-2.7 numpy

我的代码工作正常,直到我将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)]

1 个答案:

答案 0 :(得分:6)

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))]