numpy 增加数组
时间: 2023-02-09 11:56:03 浏览: 229
Numpy 改变数组维度的几种方法小结
numpy 可以使用函数 `numpy.concatenate()` 来增加数组。
示例:
```python
import numpy as np
# 定义两个数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
# 使用 concatenate 函数增加数组
c = np.concatenate((a, b), axis=0)
print(c) # 输出 [[1 2] [3 4] [5 6] [7 8]]
```
如果想沿着列增加数组,可以将 `axis` 参数设置为 `1`:
```python
import numpy as np
# 定义两个数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
# 使用 concatenate 函数增加数组
c = np.concatenate((a, b), axis=1)
print(c) # 输出 [[1 2 5 6] [3 4 7 8]]
```
注意,要使用 `concatenate` 函数增加数组,需要保证数组的形状相同(除了维度 `axis`)。
此外,numpy 还提供了一些其他用于增加数组的函数,如 `numpy.vstack()` 和 `numpy.hstack()`。这些函数可以更方便地增加数组,但是在一些情况下可能不够灵活。
阅读全文