怎么使用Numpy創(chuàng)建三維矩陣

小億
114
2024-01-25 12:11:21
欄目: 編程語言

要使用NumPy創(chuàng)建三維矩陣，可以使用numpy.array函數(shù)并傳遞一個(gè)三維數(shù)組作為參數(shù)。以下是一個(gè)示例代碼：

import numpy as np

# 創(chuàng)建一個(gè)3行2列2深度的三維矩陣
matrix = np.array([[[1, 2],
                   [3, 4],
                   [5, 6]],
                  
                  [[7, 8],
                   [9, 10],
                   [11, 12]]])

print(matrix)

輸出結(jié)果為：

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

 [[ 7  8]
  [ 9 10]
  [11 12]]]

這樣就創(chuàng)建了一個(gè)3行2列2深度的三維矩陣。

0 踩