numpy中每一行的特定列

Question

我想爲下面每行的特定列編號。 d是類似於(embedding_id x n_vocab)和t的數據是正確embedding_id的列表。

然後，我可以創建result像下面一樣。數據很好，但我認爲這並不聰明。如何以其他智能方式創建result？

t = np.random.randint(10, size=(32,))           
d = np.random.randn(30, 32)              

result = []                  
for a,b in zip(d.transpose(), t):            
    print(a[b])                 
    result.append(a[b]) # I don't think this is good way             
result = np.array(result).astype(float)           
print(result)                 
print(result.shape) # (32,). 

Answer 1

你可以這樣index;使用numpy.arrange創建列索引，並與t使用它（這就是所謂的高級索引）：

d[t, np.arange(len(t))] 

---

(d[t, np.arange(len(t))] == result).all() 
# True