通过索引向量将行从矩阵到列存储在另一个矩阵中

Question

假设我有矩阵A，零矩阵B和索引向量i，idx：

A = np.array([[1, 1, 2], [0, 0, 1]])
B = np.array([[0, 0], [0, 0], [0, 0]])
i = np.arange(len(idx))    
idx = np.array([1, 0])

通过i和idx，我知道我需要将A的 0行存储到B的1列，因为：

i = [0, 1]  #Rows at A
idx = [1, 0] # Columns at B

我需要矩阵B为：

B =
0 1 
0 1
1 2

我尝试过：

B[:, idx] = A[i, :]

，但没有用。 有人可以帮忙吗？

谢谢！ 没事

Answer 1

您已经接近建议的解决方案：

B[:,idx] = A[i,:].T