numpy数组中的奇怪赋值

时间:2014-09-10 21:28:45

标签: python arrays numpy indexing

我有一个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不会全局更改。在这种情况下我该怎么办?或者我做错了什么?

2 个答案:

答案 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显示了这一事实。