如何將csv文件寫入特定的文件夾

Question

我正在嘗試將多個.csv文件寫入一個特定目錄

這是我的代碼

with open(f+'.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)

    writer.writerow(["index", "B", "G", "R"])

    for row in rows:
        writer.writerow(row)

    writer.writerow(["Mean", mean_b/total_b, mean_g/total_g, mean_r/total_r])
    writer.writerow("STD", np.sqrt(var_b/total_b), np.sqrt(var_g/total_g), np.sqrt(var_r/total_r))

我已經將csv文件創建到與.py文件相同的目錄中，但是我想創建一個目錄並在其中創建我的csv文件

我知道我需要os.makedirs（）函數

但是我不知道我是否必須先創建目錄並指定csv文件的路徑，或者我只是將目錄名放入open（）函數中

請幫我

Answer 1

我只會創建目錄，但目錄存在錯誤

try:
   os.mkdir("./CSV")
except OSError as e:
   print("Directory exists")

with open("./CSV/" + f + ".csv", newline="") as csvfile:
   [...]

Answer 2

您可以在open語句之前添加對目錄的檢查

dir_path = 'folder_to_save_csv_file_in'
if not os.path.isdir(dir_path):
    os.makedirs(dir_path)

with open('{file_path}.csv'.format(file_path=os.path.join(dir_path, file_name), 'w+') as csv_file:
    ....

Answer 3

建議不要使用os，而建議使用pathlib模塊。 您可以使用以下命令創建目錄：

path = Path('path/to/dir')
path.mkdir(parents=True)

創建目錄及其所有缺少的父目錄。 完成此操作后，您可以使用以下命令在新目錄中創建文件

fpath = (path / 'filename').with_suffix('.csv')
with fpath.open(mode='w+') as csvfile:
    # your csv writer code