我有一个值矩阵和一个索引数组，我想遍历矩阵的每一行，只取与索引数组中的索引相对应的值。
因此，对于矩阵中的第 1 行，我取与数组中第一个元素对应的值。
Mat = [[0. 4. 4. 4.]
[2. 8. 8. 2.]
[7. 7. 7. 7.]
...
[5. 5. 5. 5.]
[7. 7. 7. 7.]
[2. 6. 6. 6.]]
arr = [0 0 3 ... 0 0 3]
所以我想要第一行的第 0 个值，第二行的第 0 个值，依此类推，最后我想将它保存在一个向量中。
np.take(Mat,arr,axis=1,out=y)
但这不起作用，我得到一个矩阵而不是向量。任何帮助，将不胜感激。
谢谢
这可以使用以下choose方法完成：
np.choose(indices, matrix.T)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句