python - 如何使用 pathlib.Path().glob() 读入文件并将具有相同文件名的文件输出到另一个子文件夹

Question

如何使用 pathlib.Path().glob() 读入文件并将具有相同文件名的文件输出到另一个子文件夹？

我的目录是这样的:

New Folder 1

-> p1_a.csv

-> p1_b.csv

-> New Folder 2

代码:

from pathlib import Path
import pandas as pd

file_path = r'C:\Users\HP\Desktop\New Folder 1'

for fle in Path(file_path).glob('p1_*.csv'):
   df = pd.read_csv(fle)

   # do something with df

   df.to_excel(file_path + r'\New Folder 2' + 'p1_*.csv' + '_new.csv')

我不确定的代码是'p1_*.csv'。

代码运行后，我的目录应该是这样的:

New Folder 1

-> p1_a.csv

-> p1_b.csv

-> New Folder 2

-> -> p1_a.csv_new.csv

-> -> p1_b.csv_new.csv

我需要在 'p1_*.csv' 位中包含什么，以便新文件夹 2 中复制的新文件具有与原始文件相同的部分文件名？

非常感谢。

Answer 1

试试这个:

import os
import pandas as pd
from pathlib import Path

source_dir = r'C:/Users/HP/Desktop/New Folder 1'

for path in Path(source_dir).glob('p1_*.csv'):
   df = pd.read_csv(path)
   # TODO: do something with df

   filename = os.path.basename(path) #--> gets the file name
   dest_path = os.path.join(source_dir, "New Folder 2", f"{filename}_new.csv")

   df.to_csv(dest_path)

现在您的目录结构如下所示:

New Folder 1
├── New Folder 2
│   ├── p1_a.csv_new.csv
│   └── p1_b.csv_new.csv
├── p1_a.csv
└── p1_b.csv