在Python中,可以使用numpy
库中的sort
函数对ndarray
进行按列排序。以下是一个示例代码:
import numpy as np
# 创建一个随机的3x3的二维数组
arr = np.random.randint(0, 10, (3, 3))
print("原始数组:")
print(arr)
# 按列对数组进行排序
sorted_arr = np.sort(arr, axis=0)
print("按列排序后的数组:")
print(sorted_arr)
运行以上代码,输出结果类似于:
原始数组:
[[8 3 2]
[6 1 9]
[4 1 7]]
按列排序后的数组:
[[4 1 2]
[6 1 7]
[8 3 9]]
在这个示例中,首先使用np.random.randint
函数创建了一个3x3的随机二维数组arr
。然后,使用np.sort
函数对数组arr
按列进行排序,指定axis=0
参数表示按列排序。最后,打印出按列排序后的数组sorted_arr
。
请注意,np.sort
函数返回的是一个新的排序后的数组,不会改变原始数组arr
的顺序。如果要在原数组上进行排序,可以使用ndarray.sort
方法,例如arr.sort(axis=0)
。
下一篇:按列对数据进行分组