I am trying to loop with some if statements but seem to be having trouble with getting the loop to 'break' if I input the wrong name. Right now the options are Chicago, New York, and Washington. I want the code to break and output 'oopsie' if I input the wrong city. Right now it does output 'oopsie' but only after I input the city and month. Any help would be greatly appreciated. I apologize for the messy code.
def common_day1(city, month):
city = input("Enter city: ")
month = int(input("Enter Month: "))
city_month = df_merged[['City','Month','Day']]
if city == "Chicago":
if month == 1:
city_month = city_month.loc[(city_month['City'] == 'CHI') & (city_month['Month'] == 1)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 2:
city_month = city_month.loc[(city_month['City'] == 'CHI') & (city_month['Month'] == 2)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 3:
city_month = city_month.loc[(city_month['City'] == 'CHI') & (city_month['Month'] == 3)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 4:
city_month = city_month.loc[(city_month['City'] == 'CHI') & (city_month['Month'] == 4)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 5:
city_month = city_month.loc[(city_month['City'] == 'CHI') & (city_month['Month'] == 5)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 6:
city_month = city_month.loc[(city_month['City'] == 'CHI') & (city_month['Month'] == 6)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif city == "New York":
if month == 1:
city_month = city_month.loc[(city_month['City'] == 'NYC') & (city_month['Month'] == 1)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 2:
city_month = city_month.loc[(city_month['City'] == 'NYC') & (city_month['Month'] == 2)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 3:
city_month = city_month.loc[(city_month['City'] == 'NYC') & (city_month['Month'] == 3)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 4:
city_month = city_month.loc[(city_month['City'] == 'NYC') & (city_month['Month'] == 4)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 5:
city_month = city_month.loc[(city_month['City'] == 'NYC') & (city_month['Month'] == 5)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 6:
city_month = city_month.loc[(city_month['City'] == 'NYC') & (city_month['Month'] == 6)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif city == "Washington":
if month == 1:
city_month = city_month.loc[(city_month['City'] == 'WAS') & (city_month['Month'] == 1)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 2:
city_month = city_month.loc[(city_month['City'] == 'WAS') & (city_month['Month'] == 2)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 3:
city_month = city_month.loc[(city_month['City'] == 'WAS') & (city_month['Month'] == 3)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 4:
city_month = city_month.loc[(city_month['City'] == 'WAS') & (city_month['Month'] == 4)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 5:
city_month = city_month.loc[(city_month['City'] == 'WAS') & (city_month['Month'] == 5)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
elif month == 6:
city_month = city_month.loc[(city_month['City'] == 'WAS') & (city_month['Month'] == 6)]
common_day = city_month['Day'].value_counts().argmax()
print("The most common day for {} during {} is {}.".format(city,month,common_day))
else:
print("oopsie")
Aucun commentaire:
Enregistrer un commentaire