NumPy中np.unique函数的用法说明

分类:知识百科 日期: 点击:0

NumPy中np.unique函数

NumPy中的np.unique函数是一个非常有用的函数,它可以用来去除数组中重复的元素,并返回一个新的数组。

使用方法

np.unique函数接受一个参数,即要处理的数组,并返回一个新的数组,其中包含去除重复元素后的元素。

# 定义一个数组
arr = np.array([1, 2, 2, 3, 3, 3, 4, 5, 5, 5, 5, 5])

# 使用np.unique函数处理数组
unique_arr = np.unique(arr)

# 输出结果
print(unique_arr)

# 输出结果
[1 2 3 4 5]

np.unique函数也可以接受一个参数,即return_index参数,当设置为True时,会返回去重后数组中每个元素在原数组中的索引。

# 定义一个数组
arr = np.array([1, 2, 2, 3, 3, 3, 4, 5, 5, 5, 5, 5])

# 使用np.unique函数处理数组,并设置return_index参数为True
unique_arr, index = np.unique(arr, return_index=True)

# 输出结果
print(unique_arr)
print(index)

# 输出结果
[1 2 3 4 5]
[0 1 3 6 7]

np.unique函数还可以接受一个参数,即return_inverse参数,当设置为True时,会返回原数组中每个元素在去重后数组中的索引。

# 定义一个数组
arr = np.array([1, 2, 2, 3, 3, 3, 4, 5, 5, 5, 5, 5])

# 使用np.unique函数处理数组,并设置return_inverse参数为True
unique_arr, inverse = np.unique(arr, return_inverse=True)

# 输出结果
print(unique_arr)
print(inverse)

# 输出结果
[1 2 3 4 5]
[0 1 1 2 2 2 3 4 4 4 4 4]

np.unique函数还可以接受一个参数,即return_counts参数,当设置为True时,会返回去重后数组中每个元素的出现次数。

# 定义一个数组
arr = np.array([1, 2, 2, 3, 3, 3, 4, 5, 5, 5, 5, 5])

# 使用np.unique函数处理数组,并设置return_counts参数为True
unique_arr, counts = np.unique(arr, return_counts=True)

# 输出结果
print(unique_arr)
print(counts)

# 输出结果
[1 2 3 4 5]
[1 2 3 1 5]

np.unique函数还可以接受一个参数,即axis参数,当设置为0时,会沿着行方向处理数组,当设置为1时,会沿着列方向处理数组。

# 定义一个数组
arr = np.array([[1,2,3],[2,3,4],[3,4,5]])

# 使用np.unique函数处理数组,并设置axis参数为0
unique_arr = np.unique(arr, axis=0)

# 输出结果
print(unique_arr)

# 输出结果
[[1 2 3]
 [2 3 4]
 [3 4 5]]

np.unique函数是NumPy中一个非常有用的函数,它可以用来去除数组中重复的元素,并返回一个新的数组。np.unique函数接受一个参数,即要处理的数组,并返回一个新的数组,其中包含去除重复元素后的元素。np.unique函数还接受三个参数,即return_index、return_inverse和return_counts参数,以及axis参数,用来控制函数的返回结果。

标签:

版权声明

1. 本站所有素材,仅限学习交流,仅展示部分内容,如需查看完整内容,请下载原文件。
2. 会员在本站下载的所有素材,只拥有使用权,著作权归原作者所有。
3. 所有素材,未经合法授权,请勿用于商业用途,会员不得以任何形式发布、传播、复制、转售该素材,否则一律封号处理。
4. 如果素材损害你的权益请联系客服QQ:77594475 处理。