溫馨提示×

np.column_stack如何用于數(shù)組合并

小樊
90
2024-06-14 15:48:29
欄目: 編程語言

np.column_stack() 函數(shù)用于將兩個或多個數(shù)組按列合并,即將數(shù)組中的元素以列的形式排列組合成一個新的數(shù)組。

例如,我們有數(shù)組 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)

輸出結(jié)果為:

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

在這個例子中,數(shù)組 A 和 B 分別有 3 個元素,通過 np.column_stack() 合并后得到了一個新的數(shù)組,該數(shù)組有 3 行 2 列,每一列分別對應(yīng)數(shù)組 A 和 B 中的元素。

0