改进np.unique不自动排序 np_unranked_unique

最新推荐文章于 2024-06-06 10:39:38 发布

原创最新推荐文章于 2024-06-06 10:39:38 发布 · 1.8k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python

Python 专栏收录该内容

36 篇文章

订阅专栏

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

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.]