Question

我希望能够检查目录中是否存在CSV文件。如果没有，我想用某些列创建一个空的数据框。如果可以，我只是想加载该文件。问题是代码读取else语句并触发未找到文件的错误。

if path.exists('raw.csv') == False:
    columns = ['join','fund_type','line_type','actual_amount','period','org']
    revenueMain = pd.DataFrame(columns=columns)
elif path.exists('raw.csv') == True:
    revenueMain = pd.read_csv('rev.csv', header=0)

Answer 1

我认为您的逻辑有误。如果第一个elif失败，则不需要if

import os.path as path

if path.exists('raw.csv'):
   revenueMain = pd.read_csv('rev.csv', header=0)
else:
   columns = ['join','fund_type','line_type','actual_amount','period','org']
   revenueMain = pd.DataFrame(columns=columns)

您还可以使用以下方法捕获错误：

try:
   revenueMain = pd.read_csv('rev.csv', header=0)
except:
   columns = ['join','fund_type','line_type','actual_amount','period','org']
   revenueMain = pd.DataFrame(columns=columns)

如果文件不存在，则创建一个数据框

1 个答案: