Ordenando elementos com NumPy

Ordenar os elementos de uma matriz é por os elementos em ordem. Por exemplo, a matriz [3, 7, 3, 4] é uma matriz desordenada. Uma matriz ordenada seria [3, 3, 4, 7]. Os objetos da ndarray da biblioteca numpy têm uma função chamada sort. A função sort recebe uma matriz desordenada e retorna uma matriz ordenada.

Exemplo

import numpy as np

array = np.array([1, 3, 5, 2, 4])
matriz_ordenada = np.sort(array)
print('Matriz não ordenada: ', array)
print('Matriz ordenada: ', matriz_ordenada)

Com o método sort também é possível ordenar outros tipos de dados, como a string.

Exemplo

import numpy as np

array = np.array(['banana', 'abacaxi', 'kiwi', 'melão'])
matriz_ordenada = np.sort(array)
print('Matriz ordenada: ', matriz_ordenada)

As strings são ordenadas seguindo a ordem do dicionário. Primeiro as palavras que começam com a, depois as que começam com b e assim por diante.
Se for passado uma matriz com duas dimensões os elementos serão ordenados com os elementos de sua dimensão. Assim os elementos das duas dimensões não se misturam. Cada dimensão é ordenada independente da outra.

Exemplo

import numpy as np

array = np.array([[2, 3, 1], [6, 5, 4]])
matriz_ordenada = np.sort(array)
print('Matriz ordenada: ', matriz_ordenada)

O mesmo vale para as matrizes com três ou mais dimensões.

Anterior Próximo