dataframe如何筛选nunique() < 10 的列
时间: 2024-03-10 18:49:25 浏览: 51
使用Python向DataFrame中指定位置添加一列或多列的方法
4星 · 用户满意度95%
可以使用`nunique()`函数来获取每一列的唯一值数量,然后使用布尔索引来筛选唯一值数量小于10的列。具体操作如下:
```python
# 假设df是一个pandas DataFrame对象
cols_to_keep = df.columns[df.nunique() < 10] # 获取唯一值数量小于10的列名
new_df = df[cols_to_keep] # 筛选出唯一值数量小于10的列
```
这样就可以得到一个新的DataFrame对象`new_df`,其中只包含唯一值数量小于10的列。
阅读全文