Question

假设我已经创建了一个数据框，并希望将其保存为数据文件夹中的 csv 文件。

我按照以下方式进行。

path = os.getcwd()
path_to_data = os.path.join(path, 'data')
df.to_csv(os.path.join(path_to_data, 'coordinates_speed.csv'))

据我所知，我的老师告诉我：

假设会有一个名为 data 的文件夹是错误的
你应该使用绝对路径

谁能告诉我正确的方法吗？

Answer 1

您可以检查data文件夹是否存在，如果不存在则将文件夹命名为data，这样您就不会假设可能没有创建data文件夹。

import os

if os.path.exists('data'):
    pass
else:
    os.mkdir('data')

df.to_csv('data/coordinates_speed.csv')

Answer 2

超快速pathlib解决方案

>>> from pathlib import Path

# append to current working directory
>>> folder = Path('.') / 'data'

# create folder if it doesn't exist
>>> folder.mkdir(exist_ok=True)

# save df
>>> df.to_csv(folder / 'coordinates_speed.csv')

如何在数据文件夹中正确保存文件

2 个答案: