如何解决同时遍历多个numpy数组
我有一个形状为(n,)的numpy数组,例如:
[
[0,1],# A
[0,1,2,3],# B
[0,2] # C
]
我该如何遍历此数组形成的所有“元组”,在这种情况下,我希望返回值为(0,0),(1,(0,3,0)...
,依此类推
解决方法
这是您的一个小小的解决方案。
inp = [
[0,1,],[0,2,3],2]
]
现在您可以简单地
import numpy as np
shp = []
for sub_list in inp:
shp.append(len(sub_list))
arr = np.ones(shp)
result = np.where(arr)
tuples = [t for t in zip(*result)]
退出:
[(0,0),(0,1),2),3,(1,2)]
此方法的工作方式是构建一个数组,数组的尺寸为列表的长度。
然后,您将获得此数组的多维索引,恰好正是您想要的。
如果您还想访问相关索引中的列表,则也可以轻松地做到这一点。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。