如何使用花哨的索引

如何使用numpy的的幻想索引创建此创建numpy的数组，我想最快的性能：如何使用花哨的索引

array([[ 1, 2, 3, 4, 16, 31], 
     [ 2, 3, 4, 5, 17, 32], 
     [ 3, 4, 5, 6, 18, 33], 
     [ 4, 5, 6, 7, 19, 34], 
     [ 5, 6, 7, 8, 20, 35], 
     [ 6, 7, 8, 9, 21, 36], 
     [ 7, 8, 9, 10, 22, 37], 
     [ 8, 9, 10, 11, 23, 38], 
     [ 9, 10, 11, 12, 24, 39], 
     [10, 11, 12, 13, 25, 40]]

与此开始：

a = np.arange(0,10) 
    aa = np.arange(0,50) 
    y = 1 
    AA = [(aa[np.array([x+y, 1+x+y, 2+x+y, 3+x+y, 15+x+y, 30+x+y])]) for x, i in enumerate(a)]

我得到这个，

[array([ 2, 3, 4, 5, 17, 32]), 
array([ 3, 4, 5, 6, 18, 33]), 
array([ 4, 5, 6, 7, 19, 34]), 
array([ 5, 6, 7, 8, 20, 35]), 
array([ 6, 7, 8, 9, 21, 36]), 
array([ 7, 8, 9, 10, 22, 37]), 
array([ 8, 9, 10, 11, 23, 38]), 
array([ 9, 10, 11, 12, 24, 39]), 
array([10, 11, 12, 13, 25, 40]), 
array([11, 12, 13, 14, 26, 41])]

来源

2017-10-12 Merlin

使用'broadcast'：'np.array（[1,2,3,4,16,31]）+ np.arange（10）[：，None]'。 – Divakar

我需要y，一个变量 - 这是一个玩具的例子..我实际上是切割'aa'，在现实世界中。 – Merlin

利用给定的变量a，aa和y的，这里有一个使用broadcasting的外加成 -

offset = np.array([0,1,2,3,15,30]) 
out = aa[a[:,None] + offset + y]

使用add-ufunc's explicit outer method -

out = aa[np.add.outer(a , offset + y)]

出界的情况下

的为出界的情况下的（AA比需要更小的），我们可以用aa垫零，然后索引到它 -

offset = np.array([0,1,2,3,15,30]) 
idx = np.add.outer(a , offset + y) 
aa_p = np.pad(aa,(0,idx.max()-len(a)+1), 'constant') 
out = aa_p[idx]

或初始化输出阵列，然后创建一个有效的分配掩码 -

offset = np.array([0,1,2,3,15,30]) 
idx = np.add.outer(a , offset + y) 
mask = idx < len(aa) 
out = np.zeros(idx.shape, dtype=aa.dtype) 
out[mask] = aa[idx[mask]]

采样输入，输出 -

In [234]: a = np.arange(0,10) 
    ...: aa = np.arange(4,42) 
    ...: y = 6 
    ...: 

In [235]: out 
Out[235]: 
array([[10, 11, 12, 13, 25, 40], 
     [11, 12, 13, 14, 26, 41], 
     [12, 13, 14, 15, 27, 0], 
     [13, 14, 15, 16, 28, 0], 
     [14, 15, 16, 17, 29, 0], 
     [15, 16, 17, 18, 30, 0], 
     [16, 17, 18, 19, 31, 0], 
     [17, 18, 19, 20, 32, 0], 
     [18, 19, 20, 21, 33, 0], 
     [19, 20, 21, 22, 34, 0]])

来源

2017-10-12 18:39:33 Divakar

我需要填充'aa'吗？如果偏移的最后一个元素大于'aa'的长度，是否会爆炸 - 或者换行。 – Merlin

@Merlin检查编辑。 – Divakar

我们可以利用广播的位置：

>>> np.arange(0,10).reshape(-1,1) + np.array([*range(1,5),16,31]) 
array([[ 1, 2, 3, 4, 16, 31], 
     [ 2, 3, 4, 5, 17, 32], 
     [ 3, 4, 5, 6, 18, 33], 
     [ 4, 5, 6, 7, 19, 34], 
     [ 5, 6, 7, 8, 20, 35], 
     [ 6, 7, 8, 9, 21, 36], 
     [ 7, 8, 9, 10, 22, 37], 
     [ 8, 9, 10, 11, 23, 38], 
     [ 9, 10, 11, 12, 24, 39], 
     [10, 11, 12, 13, 25, 40]])

这里我们创建一个范围从0到（不包括）10的矩阵，我们创建一个数据[1,2,3,4,16,31]的矩阵。

如果你想y是“偏移”，你可以写为：

>>> y = 1 
>>> np.arange(y,y+10).reshape(-1,1) + np.array([*range(0,4),15,30]) 
array([[ 1, 2, 3, 4, 16, 31], 
     [ 2, 3, 4, 5, 17, 32], 
     [ 3, 4, 5, 6, 18, 33], 
     [ 4, 5, 6, 7, 19, 34], 
     [ 5, 6, 7, 8, 20, 35], 
     [ 6, 7, 8, 9, 21, 36], 
     [ 7, 8, 9, 10, 22, 37], 
     [ 8, 9, 10, 11, 23, 38], 
     [ 9, 10, 11, 12, 24, 39], 
     [10, 11, 12, 13, 25, 40]])

来源

2017-10-12 18:36:48

如何使用花哨的索引

回答

相关问题