为什么双层数numpy数组的赋值不起作用？

Question

为什么下面的行不能像我所期望的那样工作？

import numpy as np 
a = np.array([0,1,2,1,1]) 
a[a==1][1:] = 3 
print a 
>>> [0 1 2 1 1] 
# I would expect [0 1 2 3 3] 

这是'错误'还是有另一种推荐的方式呢？

在另一方面，下面的工作：

a[a==1] = 3 
print a 
>>> [0 3 2 3 3] 

干杯，菲利普

Answer 1

看来你根本无法通过这样的双层进行分配。

这工作虽然：

a[numpy.where(a==1)[0][1:]] = 3 

Answer 2

由于一个并[a == 1]的部分是不实际的切片。它创建一个新的数组。当你考虑它的时候是有道理的 - 你只需要满足布尔条件的元素（如过滤操作）。

Answer 3

这你想要做什么

a[2:][a[2:]==1]=3 

Answer 4

它涉及到索引多么花哨的作品。有一个彻底的解释here。这样做是为了允许用花式索引进行就地修改（即a[x>3] *= 2）。这样做的结果是，你不能像你找到的那样分配给双重索引。花式索引总是返回一个副本而不是一个视图。