如何在不进行双重排序的情况下对 NumPy 数组中的元素进行有效排序？

Python/NumPy 中的高效数组排序，无需双重排序

在数据分析和机器学习中，对数组中的项目进行排序是一种常见操作。但是，对数组进行两次排序来实现此目的效率很低，因为它会增加计算复杂性。下面是使用 NumPy 的更优化方法：

要对数组中的项目进行排名而不需要两次排序，请按照以下步骤操作：

1. 应用 argsort 一次即可获取数组元素的顺序。
2. 再次使用 argsort 对步骤 1 中获得的排序进行排序。

例如：

<code class="python">import numpy as np

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

order = array.argsort()
ranks = order.argsort()

print("Original Array:", array)
print("Ordering:", order)
print("Ranks:", ranks)</code>

输出：

Original Array: [4 2 7 1]
Ordering: [3 1 2 0]
Ranks: [2 1 3 0]

As您可以看到，ranks 数组提供了原始数组中每个元素的排名，而不需要双重排序。

请注意，对于 2D 或更高维数组，指定用于排序的正确轴至关重要为 argsort 提供一个轴参数。

以上是如何在不进行双重排序的情况下对 NumPy 数组中的元素进行有效排序？的详细内容。更多信息请关注PHP中文网其他相关文章！