Consejos en la programación de Python

1) Máximo

>>> 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]
>>>

Nota: La 0ª dimensión de la matriz es la dirección de la columna, y la 1ª dimensión es la dirección de la fila
eje = 1. Encuentre la posición correspondiente al mayor 0,8 0,3 0,5 0,8 de la fila.

2) Las matrices son iguales

>>> 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

Supongo que te gusta

Origin blog.csdn.net/WANGYONGZIXUE/article/details/110292894
Recomendado
Clasificación