假设我有一个大小为m×n的矩阵A和一个大小为m×1的向量。我想使用向量中每个元素在相应行中作为偏移量来提取矩阵A中的元素。
例如,
例如,
A = [[3, 0, 0, 8, 3],
[9, 3, 2, 2, 6],
[5, 5, 4, 2, 8],
[3, 8, 7, 1, 2],
[3, 9, 1, 5, 5]]
和一个向量
y = [4, 2, 1, 3, 2]
我想实现的是一种提取A中元素的方法,使得向量中的每个元素都对应于A中相应行中的一个元素,即实现:for i in range(len(y)):
A[i, y[i]] = #perform operations here
不使用任何显式循环。
期望的输出是,
[3, 2, 5, 1, 1]
我正在使用Python和NumPy库。