开发者社区 问答 正文

在numpy.array中查找唯一的行

我需要在中找到唯一的行numpy.array。

例如:

a # I have array([[1, 1, 1, 0, 0, 0], [0, 1, 1, 1, 0, 0], [0, 1, 1, 1, 0, 0], [1, 1, 1, 0, 0, 0], [1, 1, 1, 1, 1, 0]]) new_a # I want to get to array([[1, 1, 1, 0, 0, 0], [0, 1, 1, 1, 0, 0], [1, 1, 1, 1, 1, 0]]) 我知道我可以创建一个set并在数组上循环,但是我正在寻找一种有效的纯numpy解决方案。我相信有一种方法可以将数据类型设置为void,然后我可以使用numpy.unique,但是我不知道如何使它工作。

展开
收起
保持可爱mmm 2020-01-16 17:09:20 419 分享 版权
1 条回答
写回答
取消 提交回答
  • 从NumPy 1.13开始,您可以简单地选择轴来选择任何N维数组中的唯一值。要获得唯一的行,可以执行以下操作:

    unique_rows = np.unique(original_array, axis=0) 问题来源于stack overflow

    2020-01-16 17:09:33
    赞同 展开评论
问答分类:
问答地址: