np.vstack和concatenate
时间: 2024-06-07 15:12:06 浏览: 150
np.vstack和np.concatenate都是用于将多个数组按照指定的轴进行连接的函数。
np.vstack是垂直(行)方向上的堆叠,也就是按照行的方向进行连接。它接受一个元组、列表或数组作为参数,然后将它们在垂直方向上堆叠起来。这意味着它将每个数组的行数相加,并保持列数不变。
例如:
```
import numpy as np
a = np.array([[1, 2, 3],
[4, 5, 6]])
b = np.array([[7, 8, 9],
[10, 11, 12]])
result = np.vstack((a, b))
print(result)
```
输出:
```
[[ 1 2 3]
[ 4 5 6]
[ 7 8 9]
[10 11 12]]
```
np.concatenate则可以在任意指定的轴上进行连接,默认为0轴(行轴)。它接受一个元组、列表或数组作为参数,并指定要沿着哪个轴进行连接。
例如:
```
import numpy as np
a = np.array([[1, 2, 3],
[4, 5, 6]])
b = np.array([[7, 8, 9],
[10, 11, 12]])
result = np.concatenate((a, b), axis=0)
print(result)
```
输出:
```
[[ 1 2 3]
[ 4 5 6]
[ 7 8 9]
[10 11 12]]
```
在这个例子中,np.vstack和np.concatenate的结果是相同的。但是,当你需要在除0轴(行轴)以外的轴上进行连接时,np.concatenate会更加灵活。
阅读全文