溫馨提示×

如何通過np.column_stack增加數(shù)組維度

小樊
85
2024-06-14 15:51:29
欄目: 編程語言

要通過np.column_stack增加數(shù)組的維度,可以先將數(shù)組轉換為列向量,然后再使用np.column_stack來合并這些列向量。下面是一個示例代碼:

import numpy as np

# 創(chuàng)建一個一維數(shù)組
arr1 = np.array([1, 2, 3])

# 將一維數(shù)組轉換為列向量
col_arr1 = arr1[:, np.newaxis]

# 創(chuàng)建另一個一維數(shù)組
arr2 = np.array([4, 5, 6])

# 將另一個一維數(shù)組轉換為列向量
col_arr2 = arr2[:, np.newaxis]

# 使用np.column_stack將兩個列向量合并成一個二維數(shù)組
result = np.column_stack((col_arr1, col_arr2))

print(result)

輸出結果為:

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

通過這種方式,我們可以使用np.column_stack增加數(shù)組的維度。

0