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.80.3 0.50.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