python-3.x - 如果文件不存在，则创建一个数据框

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)