用numpy用两种方法创建由1到9数字组成的3×3的二维数组
时间: 2024-02-25 13:52:03 浏览: 240
利用numpy实现一、二维数组的拼接简单代码示例
好的,以下是两种使用 NumPy 创建由1到9数字组成的3×3的二维数组的示例代码:
方法一:使用 `reshape` 函数生成二维数组
```python
import numpy as np
# 使用 reshape 函数生成二维数组
arr = np.arange(1, 10).reshape((3, 3))
# 输出二维数组
print(arr)
```
输出结果如下:
```
[[1 2 3]
[4 5 6]
[7 8 9]]
```
方法二:使用 `array` 函数和列表生成器生成二维数组
```python
import numpy as np
# 使用 array 函数和列表生成器生成二维数组
arr = np.array([[j + 1 + i * 3 for j in range(3)] for i in range(3)])
# 输出二维数组
print(arr)
```
输出结果如下:
```
[[1 2 3]
[4 5 6]
[7 8 9]]
```
这两种方法都可以生成由1到9数字组成的3×3的二维数组,你可以根据自己的需求选择合适的方法。
阅读全文