介绍unique()函数的用法
时间: 2023-06-20 12:08:49 浏览: 66
在Python中,unique()函数不是内置函数。如果你想使用这个函数,需要先导入NumPy库,因为它是NumPy库中的一个函数。unique()函数可以用来查找数组或列表中的唯一元素,并返回一个新的数组或列表,其中只包含原始数组或列表中的唯一元素。
下面是unique()函数的基本语法:
```python
numpy.unique(array, return_index=False, return_inverse=False, return_counts=False, axis=None)
```
参数说明:
- array:必需,输入的数组或列表。
- return_index:可选,如果为True,则返回输入数组或列表中唯一元素的索引位置。
- return_inverse:可选,如果为True,则返回输入数组或列表中唯一元素的索引位置的反向索引(即原始数组或列表中每个元素对应的唯一元素的索引位置)。
- return_counts:可选,如果为True,则返回输入数组或列表中每个唯一元素的出现次数。
- axis:可选,指定要查找唯一元素的轴。默认为None,表示查找整个数组或列表中的唯一元素。
下面是一个示例:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 3, 2, 1, 5])
unique_arr = np.unique(arr)
print(unique_arr)
```
这个代码会输出以下内容:
```
[1 2 3 4 5]
```
这里的unique_arr就是原始数组arr中的唯一元素。
阅读全文