我有一个n行的数组A,其中n行为3行。每行由三个整数组成,每个整数都是一个整数,它引用numpy数组中的另一个位置。例如,如果我想要N[4]
引用的行,我使用N[N[4]]
。目测:
N = np.array([[2, 3, 6], [12, 6, 9], [3, 10, 7], [8, 5, 6], [3, 1, 0] ... ])
N[4] = [3, 1 ,0]
N[N[4]] = [[8, 5, 6]
[12, 6, 9]
[2, 3, 6]]
我正在构建一个修改N的函数,我需要为某些指定的x修改N [N [x]],这也是一个参数(示例中为4)。我想将子阵列中的所有6改为另一个数字(让我们说0),所以我使用numpy.where来查找索引,这是
where_is_6 = np.where(N[N[4]] == 6)
现在,如果我像N[N[4]][where_is_6] = 0
一样直接替换,则没有任何变化。如果我之前的引用如var = N[N[4]]
然后var[where_is_6]
,则更改已完成,但本地更改为函数,并且N不会全局更改。在这种情况下我该怎么办?或者我做错了什么?
答案 0 :(得分:6)
听起来你只需要将索引转换为原始的N
坐标:
row_idxs = N[4]
r,c = np.where(N[row_idxs] == 6)
N[row_idxs[r],c] = 0
答案 1 :(得分:4)
问题是N[N[4]]
是一个新数组,您可以检查:
print(N[N[4]].flags)
C_CONTIGUOUS : True
F_CONTIGUOUS : False
OWNDATA : True
WRITEABLE : True
ALIGNED : True
UPDATEIFCOPY : False
OWNDATA
显示了这一事实。