Чтение нескольких CSV-файлов и конкатов к нескольким фреймам данных на основе имен файлов python

Question

У меня есть список csv с такими же столбцами. Вот как выглядит список:

C:/Users/foo/bar/January01.csv
C:/Users/foo/bar/February01.csv
C:/Users/foo/bar/March01.csv
C:/Users/foo/bar/January02.csv
C:/Users/foo/bar/March02.csv

Я хочу что-то вроде этого, все CSV, которые начинаются с января, должны копировать данные в январьский фрейм данных, а также за все месяцы.

Может кто-нибудь помочь мне в этом?

Answer 1

Вы можете сначала выполнить итерацию по своему каталогу, чтобы найти все месяцы, которые у вас есть, затем вы снова проходите добавление фреймов данных и, наконец, сохраняете их:

import os
dir_name = #your dir
months = set()
for file in os.listdir(dir_name):
    months.add(file[:-2])

month_df = {month: pd.DataFrame() for month in months}
for file in os.listdir(dir_name):
    month_df[file[:-2]] = month[file[:-2]].append(pd.read_csv(os.join.path(dir_name, file)))

for month in month_df.keys():
    month_df[month].to_csv(month + '.csv', index=False)