numpy矩阵最小应用于第二阵列

时间:2015-08-24 09:40:44

标签: python numpy

我真的想避免为这个简单的问题循环...

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数组中的条目。由于我只得到每列的索引,我不知道如何在没有循环或创建列表的情况下正确应用这个...谢谢!

0 个答案:

没有答案