首页 > 解决方案 > numpy:具有重复值的ndarray索引

问题描述

考虑我有一个数组:

arr2 = np.array([10., 10., 10.,  10.,  10.,  10., 10., 10., 10., 10.])

和具有重复值的索引:

idx = np.array([3,3,3,4,4,5])

如果我这样做:

arr2[idx] -=1

然后我得到了这个:

array([10., 10., 10.,  9.,  9.,  9., 10., 10., 10., 10.])

但我想减去索引中的每个值,考虑到重复!如何得到这个?

array([10., 10., 10.,  7.,  8.,  9., 10., 10., 10., 10.])

标签: pythonnumpyindexing

解决方案


推荐阅读