SZanlongo
6/24/2017 - 11:25 PM

Pickles to CSV

Pickles to CSV

import csv
import os
import pickle

def pickles_to_csv():
  with open('data_file.csv', 'w') as file:
      for filename in os.listdir("."):
          if filename.endswith(".pkl"):
              data_pickle = pickle.load(open(filename, "rb"))
              w = csv.writer(file)
              row_data = [data_pickle['key_1'], data_pickle['key_2'], data_pickle['key_3']]
              w.writerow(row_data)