我有一个二维 numpy 数组。我正在寻找每行中特定元素第一次出现的索引。输出将是一个 (n x 2) 数组，其中 n 是行数，每个条目都包含该特定元素第一次出现的 x 和 y 坐标。
非常感谢。
请您参考如下方法:
我不确定您所说的“s 和 y 坐标”到底是什么意思，所以我假设您指的是行和列的位置。
import numpy as np
np.array([(s, list(row).index(your_element)) for s,row in enumerate(your_array)])
请注意，如果 your_element 未包含在某行中，它将引发 ValueError。
以下版本将为您提供一个输出，其包含的行数可能少于输入，但不会在行中缺少 your_element 的情况下引发 ValueError .
np.array([(s, list(row).index(your_element)) for s,row in enumerate(your_array) if your_element in row])
