Como o parâmetro de eixo do NumPy funciona?
Alguém pode explicar exatamente o que oaxis
parâmetro em NumPy faz?
Estou terrivelmente confuso.
Estou tentando usar a funçãomyArray.sum(axis=num)
No começo eu pensei que se o array tivesse 3 dimensões,axis=0
retornará três elementos, consistindo na soma de todos os itens aninhados na mesma posição. Se cada dimensão contivesse cinco dimensões, eu esperavaaxis=1
para retornar um resultado de cinco itens e assim por diante.
No entanto, este não é o caso, e a documentação não faz um bom trabalho me ajudando(eles usam uma matriz 3x3x3, por isso é difícil dizer o que está acontecendo)
Aqui está o que eu fiz:
>>> e
array([[[1, 0],
[0, 0]],
[[1, 1],
[1, 0]],
[[1, 0],
[0, 1]]])
>>> e.sum(axis = 0)
array([[3, 1],
[1, 1]])
>>> e.sum(axis=1)
array([[1, 0],
[2, 1],
[1, 1]])
>>> e.sum(axis=2)
array([[1, 0],
[2, 1],
[1, 1]])
>>>
Claramente, o resultado não é intuitivo.