假设有一列数据为fruit,按照自定义分类顺序apple, banana, cherry,删除fruit列中分类不符合的行。 代码示例:
import pandas as pd
df = pd.DataFrame({'fruit':['banana', 'apple', 'orange', 'cherry', 'pear']})
#自定义分类顺序 categories = ['apple', 'banana', 'cherry']
#使用Categorical数据类型 df['fruit'] = pd.Categorical(df['fruit'],categories=categories,ordered=True)
#筛选符合条件的行 df = df[df['fruit'].notnull()]
print(df) #输出
#如果要重置索引 df = df.reset_index(drop=True)
下一篇:按照自定义规则对元素进行排序