我真的想避免为这个简单的问题循环...
import numpy as np
x = np.array([[1,2,3,4], [5,6,1,2], [7,4,9,1]])
y = np.array([[2,5,6,7], [1,2,3,4], [1,2,3,4]])
print(x)
[[1 2 3 4]
[5 6 1 2]
[7 4 9 1]]
maxidx = np.argmax(x, axis=0)
print(maxidx)
[2 1 2 0]
到目前为止一切顺利。现在我想要这些索引的y数组中的条目。由于我只得到每列的索引,我不知道如何在没有循环或创建列表的情况下正确应用这个...谢谢!