Skip to main content

Pandas - File IO

import pandas as pd

def save_df_to_csv(df, file_path):
df.to_csv(file_path, index=False)

def load_df_from_csv(file_path):
return pd.read_csv(file_path)

def save_df_to_tsv(df, file_path):
df.to_csv(file_path, index=False, sep='\t')

def load_df_from_tsv(file_path):
return pd.read_csv(file_path, sep='\t')

def add_columns_if_not_exists(df, columns):
"""
add_columns_if_not_exists(df, ["col1", "col2", "col3"])
"""
for column in columns:
if column not in df.columns:
df[column] = None
return df


def clean_columns(df, columns):
"""
clean_columns(df, ["col1"])
"""
for col in columns:
df[col] = ""

return df