不能使用NumPy为双切片赋值 [英] Cannot assign values to a 'double slice' using numpy

问题描述

如何在NumPy中执行以下操作？

1. 选择包含50%0以上值的数组的所有行。
2. 从满足1的所有行中选择第一个n(假设为2)行。
3. 做些什么，并将修改后的行放在形状相同的零数组的相同索引上。

以下结果会产生一个数组，其中不会分配新值：

In [177]:    
a = np.array([[0,0,3],[4,5,6],[7,0,0],[10,11,12],[13,14,15]])
b = np.zeros_like(a)
a
Out[177]:    
array([[ 0,  0,  3],
       [ 4,  5,  6],
       [ 7,  0,  0],
       [10, 11, 12],
       [13, 14, 15]])

In [178]:
# select all rows containg note more than 50% 0 values
percent = np.sum(a == 0, axis=-1) / float(check.shape[1])
percent = percent >= 0.5
slice = np.invert(percent).nonzero()[0]

In [183]:
# select first two rows satisfying 'slice'
a[slice][0:2] 

Out[183]:    
array([[ 4,  5,  6],
       [10, 11, 12]])

In [182]:
# do something and place modified rows on same index of zero array
b[slice][0:2] = a[slice][0:2] * 2

In [184]:
b
Out[184]:    
array([[0, 0, 0],
       [0, 0, 0],
       [0, 0, 0],
       [0, 0, 0],
       [0, 0, 0]])

推荐答案

问题是b[slice]创建的是副本而不是视图(它会触发花哨的索引)。代码b[slice][0:2]创建此副本的视图(不是原始b！)。因此..。

b[slice][0:2] = a[slice][0:2] * 2

...正在将a的相应行分配给b的副本的视图。

因为它可能导致这些情况，所以最好不要以这种方式链接索引操作。相反，只需先计算slice的相关行数，然后执行赋值：

slice = np.invert(percent).nonzero()[0][:2] # first two rows
b[slice] = a[slice] * 2

这篇关于不能使用NumPy为双切片赋值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！