numpy中.unique()函数有什么用
时间: 2023-11-27 15:54:34 浏览: 88
numpy库函数使用说明
numpy中的.unique()函数可以用于获取给定数组中的唯一元素。该函数返回一个由唯一元素组成的一维数组,并可以选择返回唯一元素的数量及其索引。
.unique()函数常用于数据预处理阶段,可用于去除重复值或标识不同的类别。 例如,可以使用.unique()函数将一个包含多个类别的数组转换为数字编码,以便于机器学习算法处理。
以下是一个示例代码:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 3, 2, 1, 3, 4, 5, 6, 6, 7, 8])
unique_values = np.unique(arr)
print(unique_values)
```
输出结果为:
```
[1 2 3 4 5 6 7 8]
```
在上面的代码中,我们使用np.unique()函数获取了给定数组中的唯一元素,并将结果存储在unique_values变量中。
阅读全文