插件窝 干货文章 pandas如何筛选某个列值是否位于某个列表内

pandas如何筛选某个列值是否位于某个列表内

class div 筛选 Yellow 103    来源:    2024-10-17

pandas筛选某个列值是否位于某个列表内

首先创建一个DataFrame:

fruits_dict = { 'Fruits':['Apple','Banana','Cherry','Dates','Eggfruit'],
                'Quantity': [5, 10, 8, 3, 7],  
                'Color': ['Red', 'Yellow', 'Red', 'Brown', 'Yellow']}
df=pd.DataFrame(fruits_dict)

观察这个DataFrame:

   Fruits      Quantity    Color
0   Apple       5           Red
1   Banana      10          Yellow
2   Cherry      8           Red
3   Dates       3           Brown
4   Eggfruit    7           Yellow

使用isin进行判断:

cond=[3,5,7]
df=df[df['Quantity'].isin(cond)]

结果:

    Fruits        Quantity    Color
0    Apple        5            Red
3    Dates        3            Brown
4    Eggfruit    7            Yellow

筛选出指定列值所对应的行 按照列值筛选数据

data = pd.read_excel("energy_consumer_statistics(按天统计能耗).xlsx")  # 1 3 7 是 预测列
data_0=data[data['电表区域'] == 0]
data_0=data_0["每天的用电量"].values

总结

以上为个人经验,希望对您有所帮助。