python - 删除 numpy 数组的重复行

Question

这个问题在这里已经有了答案:

Find unique rows in numpy.array (20 个回答)

关闭 7年前。

如何删除二维 numpy 数组的重复行？

data = np.array([[1,8,3,3,4],
                 [1,8,9,9,4],
                 [1,8,3,3,4]])

答案应该是这样的:

ans = array([[1,8,3,3,4],
             [1,8,9,9,4]])

如果有两行相同，那么我想删除一个“重复”行。

Answer 1

您可以使用 numpy unique。由于您想要唯一的行，我们需要将它们放入元组中:

import numpy as np

data = np.array([[1,8,3,3,4],
                 [1,8,9,9,4],
                 [1,8,3,3,4]])

只需将 np.unique 应用于 data 数组将导致:

>>> uniques
array([1, 3, 4, 8, 9])

打印出列表中的唯一元素。因此将它们放入元组会导致:

new_array = [tuple(row) for row in data]
uniques = np.unique(new_array)

哪个打印:

>>> uniques
array([[1, 8, 3, 3, 4],
       [1, 8, 9, 9, 4]])

更新

在新版本中，需要设置np.unique(data, axis=0)