Python中np.unique()的具体用法
时间: 2024-09-11 15:02:30 浏览: 83
python numpy之np.random的随机数函数使用介绍
在Python的NumPy库中,`np.unique()`函数主要用于找出数组中唯一的元素,并返回这些元素及其对应的索引。它有几种用法:
1. **基本用法**:
```python
unique_values = np.unique(array)
```
这将返回数组中独一无二的元素组成的数组,原数组中的重复元素会被移除。
2. **返回唯一元素及频数**:
```python
values, counts = np.unique(array, return_counts=True)
```
这会返回两个数组:一个是独特的元素,另一个是每个元素在原数组中出现的次数。
3. **保持排序**:
```python
sorted_unique, indices = np.unique(array, return_index=True)
```
`sorted_unique`包含了按顺序排列的独特元素,`indices`则给出了原数组中每个元素相对于独特元素的新位置。
4. **区分大小写**或**忽略大小写**:
如果数组元素是字符串,并希望区分大小写,就直接使用`unique()`. 若要忽略大小写,可以先转换所有元素为小写(大写)再求解:
```python
array = array.lower()
unique_values = np.unique(array)
```
阅读全文