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
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持。
