Как загрузить много файлов CSV в папку, используя Python? - PullRequest
Новая
       6

Как загрузить много файлов CSV в папку, используя Python?

0 голосов
/ 16 января 2020

Я новичок в python коде и начал замечать эту pickle функцию в python. Я пытаюсь загрузить все (50) csv файлы в папку и сохранить их как pickle файлы. Файлы csv могут содержать одинаковые или разные имена столбцов. Любые предложения о том, как подойти к этому.

1 Ответ

1 голос
/ 16 января 2020

Вы можете попробовать что-то вроде этого: 

import glob, os
import pandas as pd
import pickle

os.chdir(r"path/to/folder")
df_list = []
for file in glob.glob("*.csv"):
    df = pd.read_csv(file)
    df_list.append(df)

with open(r'\df_list.pickle', 'wb') as handle:
    pickle.dump(df_list, handle, protocol=pickle.HIGHEST_PROTOCOL)
...