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

目录

• pandas筛选某个列值是否位于某个列表内
• 筛选出指定列值所对应的行 按照列值筛选数据
• 总结

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

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持。