np.unique()的用法
时间: 2023-10-09 11:06:20 浏览: 79
`np.unique()`函数用于获取数组中的唯一元素,并以排序的方式返回。
以下是`np.unique()`函数的用法示例:
```python
import numpy as np
arr = np.array([2, 1, 3, 2, 3, 4, 5, 4, 6])
# 使用 np.unique() 函数获取数组中的唯一元素
unique_elements = np.unique(arr)
print(unique_elements)
```
输出:
```
[1 2 3 4 5 6]
```
在上述示例中,我们创建了一个一维数组`arr`。通过调用`np.unique(arr)`函数,我们获取了数组`arr`中的唯一元素。这里`unique_elements`是一个包含唯一元素的有序数组。
你还可以使用`return_counts=True`参数来获取每个唯一元素出现的次数:
```python
import numpy as np
arr = np.array([2, 1, 3, 2, 3, 4, 5, 4, 6])
# 使用 np.unique() 函数获取数组中的唯一元素及其出现次数
unique_elements, counts = np.unique(arr, return_counts=True)
print(unique_elements)
print(counts)
```
输出:
```
[1 2 3 4 5 6]
[1 2 2 2 1 1]
```
在这个例子中,除了返回唯一元素的有序数组外,还返回了每个唯一元素出现的次数。这里的`counts`是一个与`unique_elements`对应的数组,表示每个唯一元素在原始数组中的出现次数。
`np.unique()`函数在处理数组时非常有用,可以帮助我们快速获取数组中的唯一元素,并进行一些相关的分析和操作。
阅读全文