温馨提示×

怎么使用Numpy创建三维矩阵

小亿
114
2024-01-25 12:11:21
栏目: 编程语言

要使用NumPy创建三维矩阵,可以使用numpy.array函数并传递一个三维数组作为参数。以下是一个示例代码:

import numpy as np

# 创建一个3行2列2深度的三维矩阵
matrix = np.array([[[1, 2],
                   [3, 4],
                   [5, 6]],
                  
                  [[7, 8],
                   [9, 10],
                   [11, 12]]])

print(matrix)

输出结果为:

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

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

这样就创建了一个3行2列2深度的三维矩阵。

0