针对酒店类型分类
类型筛选,用户选择酒店类型(豪华型,高档型,舒适型,经济型,民宿),也可以不选择类型,那么不进行筛选
规定豪华型=5,高档型=4,舒适型=3,经济型=2,民宿=1,全部=0
输入酒店矩阵为:[酒店名称,评分,评论数,类型]
def typeSelecting(type,hotels):
results=[]
if(type==0):
return hotels
elif(type==1):
#将所有民宿提取出来
for i in range(len(hotels)):
if(hotels[i][3]=='民宿'):
results.append(hotels[i])
elif(type==2):
#将所有经济型酒店提取出来
for i in range(len(hotels)):
if(hotels[i][3]=='经济型'):
results.append(hotels[i])
elif(type==3):
#将所有舒适型酒店提取出来
for i in range(len(hotels)):
if(hotels[i][3]=='舒适型'):
results.append(hotels[i])
elif(type==4):
#将所有高档型酒店提取出来
for i in range(len(hotels)):
if(hotels[i][3]=='高档型'):
results.append(hotels[i])
elif(type==5):
#将所有豪华型酒店提取出来
for i in range(len(hotels)):
if(hotels[i][3]=='豪华型'):
results.append(hotels[i])
return results