根据值以特定顺序迭代 numpy 数组

Question

我想遍历一个 numpy 数组，从最大值的索引开始一直到最小值

import numpy as np #imports numpy package

elevation_array = np.random.rand(5,5) #creates a random array 5 by 5

print elevation_array # prints the array out

ravel_array = np.ravel(elevation_array)
sorted_array_x = np.argsort(ravel_array)
sorted_array_y = np.argsort(sorted_array_x)

sorted_array = sorted_array_y.reshape(elevation_array.shape)

for index, rank in np.ndenumerate(sorted_array):
    print index, rank

我希望它打印出来：

最高值的索引 下一个最高值的索引 下一个最高值的索引等

Answer 1

如果你想让 numpy 完成繁重的工作，你可以这样做：

>>> a = np.random.rand(100, 100)
>>> sort_idx = np.argsort(a, axis=None)
>>> np.column_stack(np.unravel_index(sort_idx[::-1], a.shape))
array([[13, 62],
       [26, 77],
       [81,  4],
       ..., 
       [83, 40],
       [17, 34],
       [54, 91]], dtype=int64)

您首先获得一个对整个数组进行排序的索引，然后将该平面索引转换为具有np.unravel_index 的索引对。对 np.column_stack 的调用只是将两个坐标数组合并为一个，并且可以用 Python 替换 zip(*np.unravel_index(sort_idx[::-1], a.shape)) 以获取元组列表而不是数组。

Answer 2

试试这个：

from operator import itemgetter

>>> a = np.array([[2, 7], [1, 4]])
array([[2, 7],
       [1, 4]])

>>> sorted(np.ndenumerate(a), key=itemgetter(1), reverse=True)
[((0, 1), 7), 
 ((1, 1), 4), 
 ((0, 0), 2), 
 ((1, 0), 1)]

如果你愿意，你可以迭代这个列表。本质上，我告诉函数sorted 根据键itemgetter(1) 对np.ndenumerate(a) 的元素进行排序。这个函数itemgetter 从((0, 1), 7), ((1, 1), 4), 生成的元组((0, 1), 7), ((1, 1), 4), 中获取第二个（索引1）元素......（即值）由np.ndenumerate(a) 生成。