python重新排列数组_python – 重新排列numpy数组中的行条目

我试图将numpy数组A转换为B而不使用循环.

A=np.array([[1,2,3],

[1,3,0],

[2,0,0]])

B=np.array([[1,2,3],

[1,0,3],

[0,2,0]])

所以在每一行中,我想使用它们的值作为索引对条目重新排序. (即在第2行,[1,3,0],1是第一个条目,3是第三个条目,0将作为第二个条目填入[1,0,3].

我可以为一行做这个,所以我可以遍历数组,但我想看看有没有办法在没有循环的情况下做到这一点.我知道循环对像这样的小数组没有什么不同,但是我担心在大型数组(1m,1m)上进行循环会产生瓶颈.

谢谢！

最佳答案 有趣的问题,1.

In [28]:

import numpy as np

A=np.array([[1,2,3],

[1,3,0],

[2,0,0]])

In [29]:

B=np.zeros(A.shape, 'int64')+np.arange(1, A.shape[0]+1)

In [30]:

np.where(np.asarray(map(np.in1d, B, A)), B, 0)

Out[30]:

array([[1, 2, 3],

[1, 0, 3],

[0, 2, 0]])

In [31]:

%timeit np.where(np.asarray(map(np.in1d, B, A)), B, 0)

10000 loops, best of 3: 156 µs per loop