如何使用 np.newaxis 在 NumPy 中操作數組維度？

理解np.newaxis：綜合指南

np.newaxis，通常表示為None，是NumPy 中的一個多功能函數，它允許您透過插入新軸來增加現有數組的維數。具體來說，它在使用一次時向數組添加一個附加維度。

使用 np.newaxis 的場景：

1。建立行/列向量：

當您想要將一維數組明確轉換為行向量（通過沿第一維插入軸）或列向量（通過沿第二個維度插入軸）。

2.廣播陣列：

np.newaxis 可以在不同維度的陣列之間執行加法等操作時使用。透過在其中一個陣列中新增一個軸，NumPy 可以啟用廣播，從而允許操作繼續進行。

3.將陣列提升到更高維度：

np.newaxis 可以多次使用將陣列提升到更高維度，有時對於涉及高階數組（張量）的複雜操作是必要的。

與np.reshape 比較：

np.newaxis 充當佔位符，暫時添加一個軸，而np.reshape 在尺寸對齊的情況下將數組重塑為指定佈局。

範例：

<code class="python"># 1D array
arr = np.arange(4)
# Create a column vector
col_vec = arr[:, np.newaxis]  # Same as arr[:, None]
print(col_vec.shape)  # (4, 1)</code>

或者，可以使用 Expand_dims 函數來提升清晰度。

<code class="python">col_vec = np.expand_dims(arr, axis=1)
print(col_vec.shape)  # (4, 1)</code>

提示：

使用 None 作為 np.newaxis 的替代品，因為它們本質上是相同的物件。

以上是如何使用 np.newaxis 在 NumPy 中操作數組維度？的詳細內容。更多資訊請關注PHP中文網其他相關文章！