要按行降序排列Numpy数组,可以使用numpy.argsort
函数来获取每行的索引,并使用这些索引来对数组进行排序。
以下是一个示例代码:
import numpy as np
# 创建一个示例数组
arr = np.array([[5, 9, 2],
[3, 7, 1],
[8, 4, 6]])
# 获取每行的索引
row_indices = np.argsort(arr, axis=1)
# 使用索引对数组进行排序
sorted_arr = np.take_along_axis(arr, row_indices, axis=1)
# 打印排序后的数组
print(sorted_arr)
输出结果为:
[[9 5 2]
[7 3 1]
[8 6 4]]
在这个示例中,我们创建了一个3x3的Numpy数组arr
,然后使用np.argsort
函数获取每行的索引。接下来,我们使用np.take_along_axis
函数根据这些索引对数组进行排序,并将排序后的数组存储在sorted_arr
变量中。最后,我们打印出排序后的数组。