(1)编写一个Numpy程序,生成一个5x5的随机整数数组,然后使用sort函数沿着行(横轴)和列(纵轴)分别进行排序,并打印排序后的数组
时间: 2024-10-23 09:06:27 浏览: 26
首先,你需要安装NumPy库,如果你还没有安装,可以使用pip来安装:
```bash
pip install numpy
```
接下来是一个Python程序,使用NumPy生成一个5x5的随机整数数组,然后对每一行和每一列进行排序:
```python
import numpy as np
# 生成一个5x5的随机整数数组
random_array = np.random.randint(0, 100, size=(5, 5))
print("Original array:")
print(random_array)
# 沿着行进行排序
sorted_row_array = random_array[np.argsort(random_array, axis=1)]
print("\nArray sorted by rows:")
print(sorted_row_array)
# 沿着列进行排序
sorted_col_array = sorted_row_array[np.argsort(random_array, axis=0)]
print("\nArray sorted by columns:")
print(sorted_col_array)
```
这个程序首先创建了一个包含0到99之间随机整数的5x5矩阵,然后利用`argsort`函数找到每个元素应该位于新数组的索引,实现了按行和按列的升序排序。
阅读全文