新闻资讯

质量为本、客户为根、勇于拼搏、务实创新

< 返回新闻资讯列表

python两个一维数组怎么合成二维数组,python两个一维数组相加

发布时间:2023-10-11 00:26:44

python两个一维数组怎样合成二维数组

要将两个一维数组合成一个二维数组,可使用numpy库中的函数np.stack()np.vstack()
使用np.stack()时,需要指定axis参数的值为0,表示按垂直方向(行)堆叠数组。
以下是示例代码:

import numpy as np
# 两个一维数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
# 使用np.stack()合成二维数组
arr_2d = np.stack((arr1, arr2), axis=0)  # axis=0表示按行堆叠
print(arr_2d)

输出结果为:

[[1 2 3]
[4 5 6]]

使用np.vstack()时,直接传入需要合并的一维数组便可。
以下是示例代码:

import numpy as np
# 两个一维数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
# 使用np.vstack()合成二维数组
arr_2d = np.vstack((arr1, arr2))
print(arr_2d)

输出结果为:

[[1 2 3]
[4 5 6]]

不管使用np.stack()np.vstack()合成二维数组,输出的结果都是相同的。