python编程 数据最大值索引_数组比较

1）最大值

>>> x=np.array([[0.1,0.8,0.1],[0.3,0.1,0.6],[0.2,0.5,0.3],[0.8,0.1,0.1]])
>>> y = np.argmax(x,axis=1)
>>> print(y)
[1 2 1 0]
>>>

注：矩阵的第0维是列方向，第1维是行方向
axis=1 从行上去找 0.8 0.3 0.5 0.8最大 分别对应的位置

2）数组相等

>>> y=np.array([1,2,1,0])
>>> t=np.array([1,2,0,0])
>>> print(y==t)
[ True  True False  True]
>>> np.sum(y==t)
3