Skip to content

Instantly share code, notes, and snippets.

@sdabbour-stratio
Created February 25, 2021 15:50
Show Gist options
  • Save sdabbour-stratio/a2f177556b8c3d5400e2d1533e623f16 to your computer and use it in GitHub Desktop.
Save sdabbour-stratio/a2f177556b8c3d5400e2d1533e623f16 to your computer and use it in GitHub Desktop.
import os
import pandas as pd
files = os.listdir(".")
csv_files = [file for file in files if '.csv' in file]
for file in csv_files:
table_name = file.lower().replace(".csv", "").replace(" ", "_")
df = pd.read_csv(file)
org_columns = df.columns.to_list()
new_columns = [column.lower().replace(" ", "_") for column in org_columns]
n = len(org_columns)
new_columns_dict = {}
for i in range(n):
new_columns_dict[org_columns[i]] = new_columns[i]
df.rename(columns=new_columns_dict, inplace=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment