如何在Python中使用os_dir正确读取CSV文件?

mmvthczy  于 2023-06-03  发布在  Python
关注(0)|答案(1)|浏览(439)

尝试用途:

os_dir = os.path.dirname(os.path.abspath(fx))
print(os_dir)

import os, inspect

if '__file__' not in locals():
    fx = inspect.getframeinfo(inspect.currentframe())[0]
else:
    fx = __file__

os_dir = os.path.dirname(os.path.abspath(fx))
print(os_dir)

我本想得到我的工作目录,但只得到了-

C:\Users\***\****Local\Temp\ipykernel_26840
6rqinv9w

6rqinv9w1#

最好的解决方案之一是使用.env文件。这些文件需要在.gitignore中被忽略,在那里,你可以放置你的敏感信息。
举个例子。
.env文件:

DATA_PATH = '/home/user/data'

my_program.py

import pandas as pd
import os
from dotenv import load_dotenv

load_dotenv()

CSV_FOLDER = os.getenv('DATA_PATH')

for file in os.listdir(CSV_FOLDER):
    print(file)
    if file.endswith('.csv'):
        pd.read_csv(file)

您甚至可以为第一个if循环添加条件,如if file.endswith('.csv') and file.startswith('data'),这样您就可以加载所需的文件。
使用.env文件将防止您将文件夹路径硬编码到代码中,并泄露敏感信息。如果你想硬编码的文件夹路径,你需要ot CSV_FOLDER = os.listdir('/home/user/data')

相关问题