python – 从多维numpy数组中选择

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 从多维numpy数组中选择，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含466字，纯文字阅读大概需要1分钟。

内容图文

我有一个多维数组a形状(nt,nz,ny,nx).尺寸是时间,z,y,x.对于x和y的每一次,我在具有形状(nt,ny,nx)的新索引数组J中选择了适当的z. J包含我想要选择的高度维度的索引.使用Python,我可以在循环中执行此操作：

b=J.copy()
for t in range(nt):
   for y in range(ny):
      for x in range(nx):
         z=J[t,y,x]
         b[t,y,x]=a[t,z,y,x]

但我想更快地完成这项工作,没有循环.这可能是微不足道的,但我无法理解它.任何人？

解决方法:

您可以将numpy.indices()与高级索引一起使用：

t, y, x = numpy.indices(J.shape)
b = a[t, J, y, x]

内容总结

以上是互联网集市为您收集整理的python – 从多维numpy数组中选择全部内容，希望文章能够帮你解决python – 从多维numpy数组中选择所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端