Numpy Sort N Dimensional Array, аналогичный списку Pythons Sort ⇐ Python

[Anonymous]

Я пытаюсь сортировать список 3D координат. Цель состоит в том, чтобы получить тот же порядок, который создает основной сортировку списка Python, который такой же, как и порядок Meshgrid + Colun_stack. Python сортирует первое значение каждого элемента сначала; В случае связей второй элемент проверяется и т. Д. Я хотел бы сделать то же самое в Numpy, особенно с функцией Argsort, поскольку он дает список индексов для отсортированного порядка. Мне нужен этот список для сортировки других массивов Numpy с тем же первым измерением. < /P>
Я думаю, что этот пост связан, но не использует Numpy для выполнения сортировки.import numpy as np

num = 10
x = np.linspace(1, 3, num)
y = np.linspace(4, 6, num)
z = np.linspace(7, 9, num)
xg, yg , zg = np.meshgrid(x, y, z, indexing='ij', sparse=False)
test = np.column_stack((xg.ravel(), yg.ravel(), zg.ravel()))
print('Target ordering', test)

np.random.shuffle(test)
print('\nShuffeld', test)

python_sorted = sorted(test.tolist())
python_sorted = np.array(python_sorted)
print('\nPython list sorted', python_sorted)

numpy_sort_idx = test.argsort(axis=0)
numpy_sorted = test[numpy_sort_idx]
print('\nNumpy arg sorted', numpy_sorted)


Подробнее здесь: https://stackoverflow.com/questions/795 ... -list-sort