vendredi 11 mai 2018

If statement trouble - python

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