dateframe只保留等于某个list的行
时间: 2023-09-09 19:11:38 浏览: 72
对Python中DataFrame按照行遍历的方法
你可以使用DataFrame的isin方法来实现这个需求。假设你有一个DataFrame对象df,想要只保留等于某个列表list_values的行,你可以按照以下方式进行操作:
```python
df_filtered = df[df['column_name'].isin(list_values)]
```
在这里,'column_name'是你要筛选的列的名称,list_values是一个包含要保留行的值的列表。
例如,如果你有一个DataFrame df,其中有一列名为'category',你想要只保留category列中值等于['A', 'B', 'C']的行,你可以这样做:
```python
list_values = ['A', 'B', 'C']
df_filtered = df[df['category'].isin(list_values)]
```
这样,df_filtered将只包含category列中值等于'A'、'B'或者'C'的行。
阅读全文