代码之家 › 专栏 › 技术社区 › Majoris

numpy数组的唯一行,同时保留其顺序

numpy python

Majoris · 技术社区 · 2 年前

如何在保留结果中行的(首次出现的)顺序的同时获得数组的唯一行?

下面的代码尝试了产生单个数组的变体。
array_equal 与元素位置进行比较。

import numpy as np
unique = np.array([])
arr = np.array([[1,2,3,4,5],[3,4,5,6,7],[5,6,7,8,9],[7,8,9,0,1],[9,0,1,2,3],[1,2,3,4,5],[-8,-7,-6,-5,-4]])

u = 0
for idx, i in enumerate(arr):
       if np.array_equal(unique, i) == False:
        unique = np.append(unique, i, axis=None)
        u += 1

print (unique)
print(u)

>>> print (unique)
[ 1.  2.  3.  4.  5.  3.  4.  5.  6.  7.  5.  6.  7.  8.  9.  7.  8.  9.
  0.  1.  9.  0.  1.  2.  3.  1.  2.  3.  4.  5. -8. -7. -6. -5. -4.]
>>> print(u)
7
>>>

对于本例,预期的结果是一个具有6个唯一行的数组。

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

1 回复 | 直到 2 年前

jared 2 年前

这可以通过简单地传递 axis=0 的论点 np.unique ,告诉numpy将这些行相互比较。要保留您可以遵循的顺序 this answer ,从而获得唯一索引,并使用这些索引(一旦排序)来选择 arr 。

import numpy as np

arr = np.array([[ 1,  2,  3,  4,  5],
                [ 3,  4,  5,  6,  7],
                [ 5,  6,  7,  8,  9],
                [ 7,  8,  9,  0,  1],
                [ 9,  0,  1,  2,  3],
                [ 1,  2,  3,  4,  5],
                [-8, -7, -6, -5, -4]])
_, idx = np.unique(arr, axis=0, return_index=True)
unique = arr[np.sort(idx)]
print(unique)

输出:

[[ 1  2  3  4  5]
 [ 3  4  5  6  7]
 [ 5  6  7  8  9]
 [ 7  8  9  0  1]
 [ 9  0  1  2  3]
 [-8 -7 -6 -5 -4]]

推荐文章

Google User · Django管理员在`list_display中未显示`creation_date`字段`

1 年前

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

1 年前

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

1 年前

user29715306 · from_users=和chats=电视节目中的差异

1 年前

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

1 年前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

1 年前

prayner · 更新嵌套字典包含列表中的项

1 年前

Bringo Jr · 我可以在O(n)中解决这个问题吗?

1 年前

Dave · 如何在for循环中修改列表值

1 年前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

1 年前