python自动填充数组
时间: 2023-12-20 16:30:10 浏览: 207
Python Numpy 自然数填充数组的实现
以下是使用NumPy库中的pad函数进行数组填充的示例代码:
```python
import numpy as np
# 创建一个3x3的数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 在数组的周围填充0,使其变成5x5的数组
padded_arr = np.pad(arr, ((1, 1), (1, 1)), 'constant')
# 输出填充后的数组
print(padded_arr)
```
运行结果为:
```
[[0 0 0 0 0 0]
[0 1 2 3 0 0]
[0 4 5 6 0 0]
[0 7 8 9 0 0]
[0 0 0 0 0 0]]
```
上述代码中,我们首先创建了一个3x3的数组arr,然后使用pad函数对其进行填充。pad函数的第一个参数是要填充的数组,第二个参数是填充的方式,这里我们使用((1, 1), (1, 1))表示在数组的周围各填充一行/列0,第三个参数是填充的方式,这里我们使用'constant'表示填充常数值0。
阅读全文