如何在Python中的2d数组中找到值的索引?
您可以np.where用来返回x和y索引数组的元组,其中给定条件保存在数组中。
np.where
如果a是阵列名称:
a
>>> np.where(a == 1) (array([0, 0, 1, 1]), array([0, 1, 2, 3]))
如果要列出(x,y)对,则可以zip使用两个数组:
zip
>>> zip(*np.where(a == 1)) [(0, 0), (0, 1), (1, 2), (1, 3)]
或者,甚至更好的是,@ jme指出这np.asarray(x).T可能是生成配对的一种更有效的方法。
np.asarray(x).T
你尚未登录,登录后可以
和开发者交流问题的细节
关注并接收问题和回答的更新提醒
参与内容的编辑和改进,让解决方法与时俱进
联系我
602392714
清零编程群