Pandas:只有在条件为Met使用函数调用时，才能将外部数据集读入Dataframe

Question

假设我在本地机器上有一个名为"test.xlsx“的Excel文件。我可以用传统的代码读取这个数据集。

df_test = pd.read_excel('test.xlsx') 

但是，如果满足条件，我想有条件地读取数据集.如果满足了另一个条件，我想在不同的数据集中阅读。

下面是我尝试使用函数的代码：

def conditional_run(x):

    if x == 'fleet':
        eval('''df_test = pd.read_excel('test.xlsx')''')

    elif x != 'fleet':
        eval('''df_test2 = pd.read_excel('test_2.xlsx')''')

conditional_run('fleet') 

下面是我遇到的错误：

File "<string>", line 1
    df_test = pd.read_excel('0Day Work Items Raw Data.xlsx')
            ^
SyntaxError: invalid syntax

Answer 1

在这种情况下，可能没有理由使用eval。根据文件的名称有条件地读取文件可能就足够了。例如：

def conditional_run(x):

    if x == 'fleet':
        file = "test.xlsx"
    elif x != 'fleet':
        file = "test_2.xlsx"
    df_test = pd.read_excel(file)
    return df_test

conditional_run('fleet')