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