怎么往numpy中添加元素
时间: 2023-11-19 12:08:59 浏览: 79
numpy向空的二维数组中添加元素的方法
可以使用`numpy.append()`函数向numpy数组中添加元素。
```python
import numpy as np
# 创建一个一维数组
arr = np.array([1, 2, 3, 4])
# 添加一个元素到末尾
arr = np.append(arr, 5)
print(arr) # 输出:[1 2 3 4 5]
```
如果要在多维数组中添加元素,需要指定添加元素的轴(axis)参数。
```python
# 创建一个二维数组
arr = np.array([[1, 2], [3, 4]])
# 沿着横轴(axis=0)添加一行
arr = np.append(arr, [[5, 6]], axis=0)
print(arr)
# 输出:
# [[1 2]
# [3 4]
# [5 6]]
```
同样的,也可以在纵轴(axis=1)上添加一列。
```python
# 沿着纵轴(axis=1)添加一列
arr = np.append(arr, [[7], [8], [9]], axis=1)
print(arr)
# 输出:
# [[1 2 7]
# [3 4 8]
# [5 6 9]]
```
注意,`numpy.append()`函数会返回一个新的数组,原始数组不会被修改。
阅读全文