Tenho uma matriz 2D a
dada por:
a = np.array([[2, 3, 1, 9], [0, 5, 4, 7], [2, 4, 6, 8]])
[[2 3 1 9]
[0 5 4 7]
[2 4 6 8]]
Gostaria de obter números de linhas de elementos máximos por coluna, ou seja, dados por np.amax(a, axis=0)
, que no meu exemplo são [0, 1, 2, 0]
. Qual é a mágica Numpy para conseguir isso?