a = np.array([[11,12,39,14], [15,16,29,18]]) b = np.array([[1,2,3,4], [5,6,7,8]])
预期结果为[5，6，3，8]我试过了，但无法解决。
result = b[range(len(a)), a.argmax(0)] print (result)
cyej8jka1#
你就快到了，你切的顺序不对：
result = b[a.argmax(0), range(a.shape[1])]
输出：array([5, 6, 3, 8])
array([5, 6, 3, 8])
1条答案
按热度按时间cyej8jka1#
你就快到了，你切的顺序不对：
输出：
array([5, 6, 3, 8])