python - 查找索引的第一次出现并保留该行

Question

这个问题已经有答案了:

how to search for unique elements by the first column of a multidimensional array (2 个回答)

已关闭 6 年前。

我有一个包含以下内容的 ndarray:

[0, 1]
[0, 5]
[1, 7]
[2, 9]
[2, 4]
[2, 4]
[3, 8]
[4, 2]
[4, 7]

现在，当多行的第一个元素相同时，我想仅保留第一行。将导致:

[0, 1]
[1, 7]
[2, 9]
[3, 8]
[4, 2]

如何使用 numpy 实现此目的？

Answer 1

给定输入数据为:

x = np.array([
    [0, 1],
    [0, 5],
    [1, 7],
    [2, 9],
    [2, 4],
    [2, 4],
    [3, 8],
    [4, 2],
    [4, 7],
])

然后您可以使用 numpy.unique 并将 return_index 设置为 true (如@divakar 在推荐中提到的)，以便找到第一个元素的唯一索引.

idx = numpyp.unique(x[:,0], return_index=True)[1]

然后您可以通过以下方式访问它们:

x[idx]

希望这有帮助。