numpy增加維度怎麼弄

numpy增加維度的方法：1、使用「np.newaxis」增加維度，「np.newaxis」是一個特殊的索引值，用於在指定位置插入一個新的維度，可以透過在對應的位置使用np.newaxis來增加維度；2、使用「np.expand_dims()」增加維度，「np.expand_dims()」函數可以在指定的位置插入一個新的維度，用於增加數組的維度

本教學作業系統：Windows10系統、Python3.11.4版本、Dell G3電腦。

在NumPy中，可以使用np.newaxis或np.expand_dims()函數來增加陣列的維度。以下將詳細介紹這兩種方法。

使用np.newaxis增加維度：

np.newaxis是一個特殊的索引值，用於在指定位置插入一個新的維度。可以透過在對應的位置使用np.newaxis來增加維度。

import numpy as np
# 创建一个一维数组
a = np.array([1, 2, 3, 4, 5])
print(a.shape)  # 输出 (5,)
# 在a的行维度上增加一个新维度
a_newaxis = a[np.newaxis, :]
print(a_newaxis.shape)  # 输出 (1, 5)
# 在a的列维度上增加一个新维度
a_newaxis = a[:, np.newaxis]
print(a_newaxis.shape)  # 输出 (5, 1)

在上述程式碼中，使用np.newaxis在原始數組a的行維度和列維度上分別增加了一個新維度。

使用np.expand_dims()增加維度：

np.expand_dims()函數可以在指定的位置插入一個新的維度，用於增加數組的維度。

import numpy as np
# 创建一个二维数组
b = np.array([[1, 2, 3], [4, 5, 6]])
print(b.shape)  # 输出 (2, 3)
# 在b的行维度上增加一个新维度
b_expand_dims = np.expand_dims(b, axis=0)
print(b_expand_dims.shape)  # 输出 (1, 2, 3)
# 在b的列维度上增加一个新维度
b_expand_dims = np.expand_dims(b, axis=1)
print(b_expand_dims.shape)  # 输出 (2, 1, 3)

在上述程式碼中，使用np.expand_dims()函數在原始數組b的行維度和列維度上分別增加了一個新維度，透過axis參數指定要在哪個維度上增加新維度。

無論是使用np.newaxis或np.expand_dims()，它們都是用來增加陣列維度的工具函數，可以根據具體需求選擇使用哪種方法來實現增加維度的目的。

以上是numpy增加維度怎麼弄的詳細內容。更多資訊請關注PHP中文網其他相關文章！