<reponame>jf20541/SpotifyETL
import pandas as pd

def clean_data():
    df = pd.read_csv('../inputs/train_api.csv')
    drop_col = ['image_url', 'is_closed', 'url', 'phone', 'display_phone','distance',
                'coordinates.latitude', 'coordinates.longitude', 'location.address2',
                'location.address3', 'location.zip_code', 'location.display_address',
                'transactions', 'location.country', 'id', 'Unnamed: 0', 'location.address1',
                'location.state']
    df = df.drop(drop_col, axis=1).reset_index()
    return df 

if __name__ == '__main__':
    clean_data()
import pandas as pd

def clean_data():
    df = pd.read_csv('../inputs/train_api.csv')
    drop_col = ['image_url', 'is_closed', 'url', 'phone', 'display_phone','distance',
                'coordinates.latitude', 'coordinates.longitude', 'location.address2',
                'location.address3', 'location.zip_code', 'location.display_address',
                'transactions', 'location.country', 'id', 'Unnamed: 0', 'location.address1',
                'location.state']
    df = df.drop(drop_col, axis=1).reset_index()
    return df 

if __name__ == '__main__':
    clean_data()