在python/pandas中,如何过滤存储在list [duplicate]中的dataframe

rsaldnfx  于 2023-06-04  发布在  Python
关注(0)|答案(1)|浏览(147)

此问题已在此处有答案

Use a list of values to select rows from a Pandas dataframe(8个回答)
5天前关闭。
在python/pandas中,如何过滤 Dataframe 列表中存储的条件?下面的代码ori_data[ori_data['category'] in ['a','d','f']]不能工作

import pandas as pd
ori_data = pd.DataFrame({'category':['a','a','b','c','d','f']})
filtered_data = ori_data[ori_data['category'] in ['a','d','f']]
e1xvtsh3

e1xvtsh31#

如果我正确理解了你的问题,你所需要做的就是使用.isin()

import pandas as pd

ori_data = pd.DataFrame({'category': ['a', 'a', 'b', 'c', 'd', 'f']})
filtered_data = ori_data[ori_data['category'].isin(['a', 'd', 'f'])]

相关问题