您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何在Python中的2d数组中找到值的索引?

如何在Python中的2d数组中找到值的索引?

您可以np.where用来返回x和y索引数组的元组,其中给定条件保存在数组中。

如果a是阵列名称

>>> np.where(a == 1)
(array([0, 0, 1, 1]), array([0, 1, 2, 3]))

如果要列出(x,y)对,则可以zip使用两个数组:

>>> zip(*np.where(a == 1))
[(0, 0), (0, 1), (1, 2), (1, 3)]

或者,甚至更好的是,@ jme指出这np.asarray(x).T可能是生成配对的一种更有效的方法

python 2022/1/1 18:37:34 有224人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶