【发布时间】:2019-01-26 15:29:48
【问题描述】:
假设我有矩阵 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, :]
它没有工作。 有人可以帮忙吗?
谢谢! 诺法尔
【问题讨论】:
标签: python numpy matrix indexing