要创建一个3D数组,其索引由一个2D数组给出,可以使用NumPy库来实现。
首先,导入NumPy库:
import numpy as np
然后,创建一个2D数组:
m = 3 # 行数
n = 4 # 列数
array_2d = np.arange(m * n).reshape(m, n)
这将创建一个形状为(m, n)的2D数组,其中元素的值为0到(m * n - 1)。
接下来,使用2D数组的形状来创建一个3D数组:
o = 5 # 第三维的长度
array_3d = np.arange(m * n * o).reshape(m, n, o)
这将创建一个形状为(m, n, o)的3D数组,其中元素的值为0到(m * n * o - 1)。
你可以根据需要调整m、n和o的值来创建不同形状的数组。
关于NumPy库的更多信息和使用方法,你可以参考腾讯云的产品介绍链接:NumPy产品介绍。
领取专属 10元无门槛券
手把手带您无忧上云