函数numpy.argmax()如何处理带有数字和字符的列表

时间:2019-03-22 17:11:21

标签: python-3.x numpy argmax

即使index = 1是字符,如何确定index = 0处具有最高值的行?

我有一个包含两个位置的列表数组,其中第一个位置是数字,第二个位置是字符。

A=np.array([[150,"A"],[9,1],[9,14]])
B=np.argmax(A, axis=0)

我应该收到这个:

[0 2]  

但是我明白了:

[1 0]

我不确定第二个位置的输出是什么,但是对于第一个位置它的输出应该为0。我对函数有什么误解?甚至切片都无法解决。

1 个答案:

答案 0 :(得分:0)

我找到了解决方案。我不知道,以后是否会遇到问题!

A=np.array([(150,"A"),(30,"B"),(9,"14")], dtype='object')
c=np.argmax(A, axis=0)