首頁  >  文章  >  後端開發  >  如何將 4D NumPy 陣列重塑為 2D 陣列?

如何將 4D NumPy 陣列重塑為 2D 陣列?

Mary-Kate Olsen
Mary-Kate Olsen原創
2024-10-31 18:09:46817瀏覽

How to Reshape a 4D NumPy Array to a 2D Array?

在NumPy 中將4D 數組重塑為2D 數組背後的直覺和想法

在NumPy 中,重塑多維數組需要了解多維數組軸排列和重塑功能。要將 4D 數組重塑為 2D 數組,我們通常遵循三個步驟:

  1. 排列軸: 重新排列數組的維度以匹配所需的輸出形狀。這是使用 transpose、moveaxis、rollaxis 或 swapaxes 等函數完成的。
  2. 分割或合併軸:透過建立新軸(分割)或組合現有軸來調整陣列的維度(
  3. 重塑:使用重塑函數將陣列的形狀調整為所需的輸出大小。 >
  4. 將其重塑為(4,4):

排列軸:

我們希望軸順序為(0, 1, 3, 2)。使用轉置來實現:

array([[[[ 0,  0],
         [ 0,  0]],

        [[ 5, 10],
         [15, 20]]],


       [[[ 6, 12],
         [18, 24]],

        [[ 7, 14],
         [21, 28]]]])

    分割軸:
  1. 不執行分割,因為我們已經有了正確的維度數。 重塑: 重塑為所需形狀:
a = a.transpose((2, 0, 3, 1))
  1. 結果:

以上是如何將 4D NumPy 陣列重塑為 2D 陣列?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn