python - 获取二维数组内元素索引的最佳方法是什么？

Question

所以我正在开发一个项目，其中有一个二维数组，并且我需要找到放置在第二维上的元素的位置，而不知道它的第一维的索引。

我知道我可以迭代这些部分，检查 list[x].index(y) 是否不是值错误，但这使我的代码不太优雅寻找，我想知道是否有更好的方法可以做到这一点，例如 2d .index 方法？

Answer 1

假设您不想重新发明轮子，我相信这个问题有您正在寻找的答案:Is there a Numpy function to return the first index of something in an array?

import numpy as np
a = np.array([[1,2,4],[4,5,6]])
item = 4
row_indices, col_indices = np.where(a == item)