如何使用花哨的索引

如何使用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

如何使用花哨的索引

回答

相關問題