np.unique
时间: 2023-07-04 07:21:48 浏览: 98
UNIQUE
`np.unique` 是 NumPy 库中的一个函数,它用于返回给定数组中的唯一元素,并按升序排列。
函数的语法如下:
```python
np.unique(arr, return_index=False, return_inverse=False, return_counts=False, axis=None)
```
参数说明:
- `arr`: 输入的数组。
- `return_index`: 如果为 True,则返回输入数组中唯一元素的索引。
- `return_inverse`: 如果为 True,则返回输入数组中唯一元素的下标。
- `return_counts`: 如果为 True,则返回输入数组中唯一元素的出现次数。
- `axis`: 用于指定沿着哪个轴计算唯一值,默认为 None。
示例:
```python
import numpy as np
arr = np.array([1, 2, 2, 3, 3, 3])
unique_arr = np.unique(arr)
print(unique_arr)
# 输出: [1 2 3]
```
阅读全文