根据给定选项读取文件

Question

我有一个程序，其中包含

定义的多个函数

def function_1():

等等。这些通过此处定义的菜单访问：

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.")

菜单如下所示：

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")

我有 3 个文件，它们的名称如下： 2000.rep.csv 2010.rep.csv 2020.rep.csv

我需要找到一种方法让程序询问 （在程序开始时） 2000、2010 或 2020 之间的一年，并让程序读取那一年的文件，使用类似的东西： df = pd.read_csv('2010.rep.csv').

如果需要，我还需要程序提供通过菜单更改年份的选项（如果需要，请阅读不同的文件）。

我已经尝试了很多东西，但我无法工作，我只需要一点帮助，因为 python 对我来说是新的。感谢任何帮助。

使用 Python 2.7 和 Pandas 0.24.2

Answer 1

做一个选项，就像你所做的那样。

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')

根据给定选项读取文件

Reading file based on given option

python

menu

file

python-2.7

pandas