python数组增加维度,新增的维度填充0
时间: 2024-03-21 14:41:21 浏览: 229
Python Numpy 自然数填充数组的实现
5星 · 资源好评率100%
如果你希望在增加维度的同时,对新增的维度进行填充,可以使用numpy库中的zeros方法创建一个全0的数组,然后使用numpy的concatenate方法将原数组和全0数组连接起来。例如,将一维数组a增加为二维数组b,并在新增的维度上填充0:
```python
import numpy as np
a = np.array([1, 2, 3, 4])
zeros = np.zeros((a.shape[0], 1), dtype=a.dtype)
b = np.concatenate((a[:, np.newaxis], zeros), axis=1)
print(b)
```
输出结果为:
```
[[1 0]
[2 0]
[3 0]
[4 0]]
```
在上面的代码中,zeros方法创建了一个全0的二维数组,其行数与原数组a的长度相同,列数为1,数据类型与原数组a相同。然后使用concatenate方法按列连接原数组和全0数组,生成了一个新的二维数组b。
阅读全文