n个一维数组组成n+1维数组,value等于索引值

以n=3为例

# 要生成的三维数组各向维度
Z, Y, X = 90, 250, 410
# 定义索引值
z = np.array(range(Z))
y = np.array(range(Y))
x = np.array(range(X))
# 各维度分别生成三维数组,value为该维度索引
z = z.reshape(-1,1).repeat(Y,axis=1)
z = np.expand_dims(z,axis=2).repeat(X,axis=2)
y = y.reshape(1,-1).repeat(Z,axis=0)
y = np.expand_dims(y,axis=2).repeat(X,axis=2)
x = x.reshape(1,-1).repeat(Y,axis=0)
x = np.expand_dims(x,axis=0).repeat(Z,axis=0)
# 拼接成所要数组
pos = np.zeros((Z,Y,X,3))
pos[:,:,:,0] = z
pos[:,:,:,1] = y
pos[:,:,:,2] = x

输出

pos[45,15,206]
Out[0]: array([ 45.,  15., 206.])

pos[45,15,26]
Out[1]: array([45., 15., 26.])

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值