np.array的元素添加
时间: 2024-05-26 11:06:49 浏览: 122
np.array的元素添加可以通过numpy库中的函数来实现。具体有以下几种方法:
1. 使用numpy.append()函数在数组末尾添加元素[^1]:
```python
import numpy as np
arr = np.array([1, 2, 3])
new_arr = np.append(arr, 4)
print(new_arr) # 输出:[1 2 3 4]
```
2. 使用numpy.insert()函数在指定位置插入元素:
```python
import numpy as np
arr = np.array([1, 2, 3])
new_arr = np.insert(arr, 1, 4)
print(new_arr) # 输出:[1 4 2 3]
```
3. 使用numpy.concatenate()函数将两个数组连接起来:
```python
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
new_arr = np.concatenate((arr1, arr2))
print(new_arr) # 输出:[1 2 3 4 5 6]
```
4. 使用numpy.vstack()函数在垂直方向上堆叠两个数组:
```python
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
new_arr = np.vstack((arr1, arr2))
print(new_arr)
# 输出:
# [[1 2 3]
# [4 5 6]]
```
5. 使用numpy.hstack()函数在水平方向上堆叠两个数组:
```python
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
new_arr = np.hstack((arr1, arr2))
print(new_arr) # 输出:[1 2 3 4 5 6]
```
阅读全文