numpy.rollaxis详细介绍

import numpy as np

arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
print("原始数组：")
print(arr)

new_arr = np.rollaxis(arr, axis=1)
print("沿着轴1滚动后的数组：")
print(new_arr)

原始数组：
[[[ 1  2  3]
  [ 4  5  6]]

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

沿着轴1滚动后的数组：
[[[ 1  2  3]
  [ 7  8  9]]

 [[ 4  5  6]
  [10 11 12]]]

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print("原始数组：")
print(arr)

new_arr = np.rollaxis(arr, axis=0, start=2)
print("在指定位置添加新的维度：")
print(new_arr.shape)

原始数组：
[[1 2 3]
 [4 5 6]]

在指定位置添加新的维度：
(1, 2, 3)