如何按特定列对 NumPy 数组进行排序？

按列对 NumPy 数组进行排序

按特定列对 NumPy 数组进行排序通常是数据操作的必要操作。例如，考虑数组：

a = array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])

假设我们要根据第二列中的值对 a 的行重新排序。为了实现这一点，我们可以利用 NumPy 中的 argsort() 函数。

a[:, 1].argsort()

上面的表达式将 argsort() 应用于 a 的第二列，生成与排序值相对应的索引数组。具体来说，它返回：

array([1, 0, 2])

这些索引表示 a 的行应重新排序以按第二列升序排序的顺序。结合数组索引，我们可以得到排序后的数组：

a[a[:, 1].argsort()]

该表达式返回：

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])

根据需要，a 的行已根据第二列排序.

以上是如何按特定列对 NumPy 数组进行排序？的详细内容。更多信息请关注PHP中文网其他相关文章！