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

在3D numpy数组中查找唯一元组

numpy python

jss367 · 技术社区 · 4 年前

我试图在numpy数组中查找uniques元组,但无法找到。基于其他的答案我试过了 np.unique 在设置轴的值时,但它没有提供我要找的东西。举个例子:

b = np.array([[[255, 0, 0], [255, 0, 0]], [[255, 0, 0], [0, 0, 0]]])

我正在寻找一种方法来告诉我它有两个元组:(255,0,0)和(0,0,0)。以下是使用 np.唯一

np.unique(b, axis=0)

array([[[255,   0,   0],
        [  0,   0,   0]],

       [[255,   0,   0],
        [255,   0,   0]]])


np.unique(b, axis=1)

array([[[255,   0,   0],
        [255,   0,   0]],

       [[  0,   0,   0],
        [255,   0,   0]]])

np.unique(b, axis=2)

array([[[  0, 255],
        [  0, 255]],

       [[  0, 255],
        [  0,   0]]])

我怎么让它回来 [255, 0, 0], [0, 0, 0] ?

2 回复 | 直到 4 年前

Frank Yellin 4 年前

先把b做成Nx3数组。然后使用unique。

>>> np.unique(b.reshape(-1, 3), axis=0)
array([[  0,   0,   0],
       [255,   0,   0]])

U13-Forward 4 年前

尝试使用 list 理解力:

print(np.array([i for i in b if len(set(map(tuple, i))) == len(i)]))

[[[255   0   0]
  [  0   0   0]]]

推荐文章

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

6 月前

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

6 月前

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

6 月前

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

6 月前

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

6 月前

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

6 月前

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

6 月前

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

6 月前

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

6 月前

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

6 月前