I have a program that has multiple functions defined by
def function_1():
And so on. These are accessed through a menu defined here:
menu()
option = int(raw_input("Select an option: "))
while option != 0:
if option == 1:
function_1(df)
elif option == 2:
function_2(df)
elif option == 3:
function_3(df)
elif option == 4:
function_4(df)
else:
print("This option is not valid.")
menu()
option = int(input("Select option: "))
print("Thanks for using program.")
The menu looks like this:
def menu():
print("[1]: 1st Function")
print("[2]: 2nd Function")
print("[3]: 3rd Function")
print("[4]: 4th Function")
print("[6]: Modify the year to analyse")
print("[0]: Quit Program")
I have 3 files that I have are named as followed:
2000.rep.csv
2010.rep.csv
2020.rep.csv
I need to find a way for the program to ask (at the start of the program) for a year between 2000, 2010 or 2020 and for the program to read the file for that year, using something like: df = pd.read_csv('2010.rep.csv').
I also need the program to offer the option to change the year through the menu if needed (so read a different file if needed).
I have tried multiple things but I can't get any to work, I just need a bit of help as python is new to me. Any help is appreciated.
Using Python 2.7 and Pandas 0.24.2
CodePudding user response:
Make an option, just like what you did.
def select_year():
print('2000')
print('2010')
print('2020')
select_year()
year = str(raw_input('Please select a year: '))
# If year is not in list
while not year in ['2000','2010','2020']:
print('Year is not valid')
select_year()
year = str(raw_input('Please select a year: '))
df = pd.read_csv(year '.rep.csv')