python - 如何从 2D numpy 数组中的每一行获取元素的第一次出现？

Question

我有一个二维 numpy 数组。我正在寻找每行中特定元素第一次出现的索引。输出将是一个 (n x 2) 数组，其中 n 是行数，每个条目都包含该特定元素第一次出现的 x 和 y 坐标。

非常感谢。

Answer 1

我不确定您所说的“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])