温馨提示×

np.column_stack如何用于数组合并

小樊
97
2024-06-14 15:48:29
栏目: 编程语言

np.column_stack() 函数用于将两个或多个数组按列合并,即将数组中的元素以列的形式排列组合成一个新的数组。

例如,我们有数组 A 和 B:

import numpy as np

A = np.array([1, 2, 3])
B = np.array([4, 5, 6])

result = np.column_stack((A, B))

print(result)

输出结果为:

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

在这个例子中,数组 A 和 B 分别有 3 个元素,通过 np.column_stack() 合并后得到了一个新的数组,该数组有 3 行 2 列,每一列分别对应数组 A 和 B 中的元素。

0