gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 17:41:08 25 4
gpt4 key购买 nike

我希望能够检查我的目录中是否存在 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)

最佳答案

我认为你的逻辑是错误的。不需要 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)

关于python-3.x - 如果文件不存在,则创建一个数据框,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54448944/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com