Numpy array 创建一维 二维 三维数组

import numpy as np
temp = np.array([0,1,2,3,4,5])  # 一维数组
print(temp)
print(temp.shape)

[0 1 2 3 4 5]
(6,)

import numpy as np
temp = np.array([[0,1,2],[3,4,5]])  # 二维数组
print(temp)
print(temp.shape)

[[0 1 2]
 [3 4 5]]
(2, 3)

import numpy as np
temp = np.array([
    [[0,1,2],[3,4,5]],
    [[6,7,8],[9,0,1]],
    [[2,3,4],[5,6,7]],
    [[8,9,0],[1,2,3]],
])  # 三维数组
print(temp)
print(temp.shape)

[[[0 1 2]
  [3 4 5]]

 [[6 7 8]
  [9 0 1]]

 [[2 3 4]
  [5 6 7]]

 [[8 9 0]
  [1 2 3]]]
(4, 2, 3)