optimization_metadata.io_dataframe

Complexity

Total Complexity

4

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	load_dataframes()	0	8	2
A	save_dataframe()	0	5	2

# Author: Simon Blanke
# Email: simon.blanke@yahoo.com
# License: MIT License


import os
import glob
import pandas as pd


def save_dataframe(path, name, _dataframe):
    if not os.path.exists(path):
        os.makedirs(path, exist_ok=True)

    _dataframe.to_csv(path + name + ".csv", index=False)


def load_dataframes(path):
    paths = glob.glob(path + "*.csv")

    dataframe_list = []
    for path in paths:
        dataframe_list.append(pd.read_csv(path))

    return dataframe_list