改进np.unique不自动排序 np_unranked_unique

Kenn7

于 2020-07-22 11:49:12 发布

阅读量1.7k

点赞数 3

分类专栏： Python 文章标签： python

本文链接：https://blog.youkuaiyun.com/kane7csdn/article/details/107509838

版权

Python 专栏收录该内容

36 篇文章

订阅专栏

本文介绍了一种使用Python的NumPy库从数组中提取唯一元素的方法。通过定义一个名为np_unranked_unique的函数，该函数接收一个NumPy数组作为输入，并返回一个包含所有不重复元素的新数组。示例代码展示了如何创建一个包含重复数字的数组，然后应用此函数来获取其唯一值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np


def np_unranked_unique(nparray):
    n_unique = len(np.unique(nparray))
    ranked_unique = np.zeros([n_unique])
    i = 0
    for x in nparray:
        if x not in ranked_unique:
            ranked_unique[i] = x
            i += 1
    return ranked_unique


arr = np.array([2, 3, 3, 1, 2])
print(np.unique(arr), np_unranked_unique(arr))

output: [1 2 3] [2. 3. 1.]