+-
如何在numpy数组中获取具有不同数量的元素的唯一元素?
我想在numpy 2D数组中获得独特性,但数组是这样的:

a = np.array([[1,2,3], [2,3], [1]])
np.unique(a)

因此,数组具有不同数量的元素,我希望扁平数组具有如下所示的唯一元素:

[1,2,3]

但“np.unique”没有按预期工作.

最佳答案
由于内部列表的长度不同,你有一个对象类型数组,np.unique会比较对象(内部列表)而不是元素;您需要在1d数组中使用np.concatenate手动压平数组,然后使用np.unique:

np.unique(np.concatenate(a))
# array([1, 2, 3])
点击查看更多相关文章

转载注明原文:如何在numpy数组中获取具有不同数量的元素的唯一元素? - 乐贴网