首先创建一个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
以上为个人经验,希望对您有所帮助。