diff --git a/csvprocess.py b/csvprocess.py index d35ea7c..fdbf588 100755 --- a/csvprocess.py +++ b/csvprocess.py @@ -137,10 +137,10 @@ def get_range_date(attr): return date[1], date[2] -def export_csv(dataframe, path): +def export_csv(dataframe, path, index = False): logger.debug('export csv from panda dataframe') try: - dataframe.to_csv(path ,header = False, index= False) + dataframe.to_csv(path ,header = False, index = index) except Exception as e: logger.error( 'Erro when exporting Dataframe to csvfile {}. \n{}'.format( @@ -283,7 +283,7 @@ if __name__ == '__main__': consolidated_stats = pd.DataFrame.from_dict(temp) t.stop() - + indicators=indicators[['indicateur']] t.start('Process_town') towns = pd.merge(towns, statistics[['CODGEO', 'SUPERF']], @@ -301,7 +301,7 @@ if __name__ == '__main__': export_csv(states, args.export + '/region.csv') export_csv(dep_capitals, args.export + '/cheflieudepartement.csv') export_csv(states_capitals, args.export + '/cheflieuregion.csv') - export_csv(indicators, args.export + '/indicateur.csv') + export_csv(indicators, args.export + '/indicateur.csv', True) export_csv(consolidated_stats, args.export + '/statistiques.csv') t.stop()