我想要轻松读取多维 numpy 数组的某些部分。对于任何数组,访问第一维都很容易(b[index]
)。另一方面,访问第六维度是“困难的”(尤其是阅读)。
b[:,:,:,:,:,index] #the next person to read the code will have to count the :
有一个更好的方法吗?
特别是有没有一种方法,在编写程序时不知道轴?
编辑:
索引维度不一定是最后一个维度
您可以使用np.take https://docs.scipy.org/doc/numpy/reference/generated/numpy.take.html。
例如:
b.take(index, axis=5)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)