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

总结

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