numpy.ndarray添加元素
时间: 2023-09-24 13:06:28 浏览: 119
np.newaxis 实现为 numpy.ndarray(多维数组)增加一个轴
5星 · 资源好评率100%
numpy.ndarray是一个多维数组对象,可以使用numpy.append()函数来添加元素。
例如,在一个一维数组中添加一个元素:
```python
import numpy as np
# 创建一个一维数组
a = np.array([1, 2, 3])
print("原数组:", a)
# 添加一个元素
a = np.append(a, 4)
print("添加后的数组:", a)
```
输出:
```
原数组: [1 2 3]
添加后的数组: [1 2 3 4]
```
在一个二维数组中添加一行或一列元素:
```python
import numpy as np
# 创建一个二维数组
a = np.array([[1, 2], [3, 4]])
print("原数组:\n", a)
# 添加一行元素
b = np.array([5, 6])
a = np.vstack((a, b))
print("添加一行后的数组:\n", a)
# 添加一列元素
c = np.array([7, 8]).reshape(2, 1)
a = np.hstack((a, c))
print("添加一列后的数组:\n", a)
```
输出:
```
原数组:
[[1 2]
[3 4]]
添加一行后的数组:
[[1 2]
[3 4]
[5 6]]
添加一列后的数组:
[[1 2 7]
[3 4 8]
[5 6 0]]
```
注意,在添加一列元素时,需要先将一维数组转换成二维数组才能使用np.hstack()函数。此外,由于numpy.ndarray是一个静态数组,添加元素时会创建一个新的数组对象,原数组并不会改变。
阅读全文