假设我有一个给定维度的矩阵,还有一个包含一定范围值的数组。数组中包含的值是我要从矩阵中提取值的行的索引。因此,一个示例可能是:
A = np.array([[3, 6, 7, 5, -3, 0], [5, -2, 2, 51, -13, 8], [13, -17, 18, 22, -12, 90], [33, -12, 12, 32, -42, 90]])
B = np.array([0, 2, 3])
预期结果:
array([[ 3, 7, 5],
[ 5, 2, 51],
[13, 18, 22],
[33, 12, 32]])
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句