Question

说我有numpy数组arr_1 = np.arange(10)返回：

array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

如何使用切片将多个元素更改为某个值？

例如：将从第一个元素开始每五个元素出现的第0个，第一个和第二个元素更改为100.我想要这样：

array([0, 100, 100, 100, 4, 5, 100, 100, 100, 9])

我尝试了arr_1[1::[5, 6, 7]] = 100，但这不起作用。

Answer 1

以下是基于您所做的另一种解决方案：

arr_1 = np.arange(10)
arr_1[1::5] = 100
arr_1[2::5] = 100
arr_1[3::5] = 100

然后它返回：

array([  0, 100, 100, 100,   4,   5, 100, 100, 100,   9])

Answer 2

如果重复偏移量除以数组长度：

a.reshape((-1, 5))[:, 1:4] = 100

一般情况需要两行：

a[: len(a) // 5 * 5].reshape((-1, 5))[:, 1:4] = 100
a[len(a) // 5 * 5 :][1:4] = 100

工作原理：以所描述的方式重新整形堆叠数组的连续延伸，使目标子基本对齐，因此可以使用标准的2d索引一次性解决：

>>> a = np.arange(15)
>>> a.reshape((-1, 5))
array([[ 0,  1x,  2x,  3x,  4],
       [ 5,  6x,  7x,  8x,  9],
       [10, 11x, 12x, 13x, 14]])

Answer 3

这是masking -

的一种方法

a = np.arange(10)       # Input array
idx = np.array([0,1,2]) # Indices to be set
offset = 1              # Offset

a[np.in1d(np.mod(np.arange(a.size),5) , idx+offset)] = 100

使用原始样本运行样本 -

In [849]: a = np.arange(10)       # Input array
     ...: idx = np.array([0,1,2]) # Indices to be set
     ...: offset = 1              # Offset
     ...: 
     ...: a[np.in1d(np.mod(np.arange(a.size),5) , idx+offset)] = 100
     ...: 

In [850]: a
Out[850]: array([  0, 100, 100, 100,   4,   5, 100, 100, 100,   9])

使用non-sequential indices -

运行示例

In [851]: a = np.arange(11)       # Input array
     ...: idx = np.array([0,2,3]) # Indices to be set
     ...: offset = 1              # Offset
     ...: 

In [852]: a[np.in1d(np.mod(np.arange(a.size),5) , idx+offset)] = 100

In [853]: a
Out[853]: array([  0, 100,   2, 100, 100,   5, 100,   7, 100, 100,  10])

Answer 4

您只需要在np.array（list）中包装索引列表。你非常接近正确：

In [2]: arr_1 = np.arange(10)

In [3]: arr_1[np.array([0,1,2,5,6,7])] = 100

In [4]: arr_1
Out[4]: array([100, 100, 100,   3,   4, 100, 100, 100,   8,   9])

我根据您的要求使用手工编码的索引值。您可以使用您喜欢的某种技术以自动方式获取索引，如Divakar所示。

使用Python中的切片更改多个Numpy数组元素

4 个答案: