温馨提示×

如何通过np.column_stack增加数组维度

小樊
86
2024-06-14 15:51:29
栏目: 编程语言

要通过np.column_stack增加数组的维度,可以先将数组转换为列向量,然后再使用np.column_stack来合并这些列向量。下面是一个示例代码:

import numpy as np

# 创建一个一维数组
arr1 = np.array([1, 2, 3])

# 将一维数组转换为列向量
col_arr1 = arr1[:, np.newaxis]

# 创建另一个一维数组
arr2 = np.array([4, 5, 6])

# 将另一个一维数组转换为列向量
col_arr2 = arr2[:, np.newaxis]

# 使用np.column_stack将两个列向量合并成一个二维数组
result = np.column_stack((col_arr1, col_arr2))

print(result)

输出结果为:

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

通过这种方式,我们可以使用np.column_stack增加数组的维度。

0