在numpy下合併兩個數組

Question

使用Numpy，我想在b和c給出下面的結果。我已經研究了堆疊功能，但是我無法使其工作。有人可以幫忙嗎？

import numpy as np 

a=range(35,135) 

b=np.reshape(a,(10,10)) 
c=np.array([[5,5],[5,6],[5,7],[6,5],[6,6],[6,7],[7,5],[7,6],[7,7]]) 

結果應該是這樣的：

np.array([[5,5,90],[5,6,91],[5,7,92],[6,5,100],[6,6,101],[6,7,102], 
      [7,5,110],[7,6,111],[7,7,112]]) 

Answer 1

唷！這是一個doosie。首先，我們使用numpy的的幻想索引拔出來，你想：

>>> b[tuple(c.T)] 
array([ 90, 91, 92, 100, 101, 102, 110, 111, 112]) 

隨後的項目，剩下的唯一的事情就是堆棧陣列背靠c使用column_stack：

>>> np.column_stack((c,b[tuple(c.T)])) 
array([[ 5, 5, 90], 
     [ 5, 6, 91], 
     [ 5, 7, 92], 
     [ 6, 5, 100], 
     [ 6, 6, 101], 
     [ 6, 7, 102], 
     [ 7, 5, 110], 
     [ 7, 6, 111], 
     [ 7, 7, 112]])