Update script.py

main
Nathan BOILEAU 3 years ago
parent 9d37b51008
commit 3d61f0d51c

@ -2,7 +2,7 @@ import pandas as pd
import psycopg2 as psy import psycopg2 as psy
data = pd.read_csv(r'911.csv') data = pd.read_csv(r'co2_emission.csv')
df = pd.DataFrame(data) df = pd.DataFrame(data)
df2 = df.drop_duplicates() df2 = df.drop_duplicates()
co = None co = None
@ -15,26 +15,22 @@ try :
password = 'azertyuiop') password = 'azertyuiop')
curs=co.cursor() curs=co.cursor()
curs.execute('''DROP TABLE IF EXISTS 911;''') curs.execute('''DROP TABLE IF EXISTS CO_DEUX;''')
curs.execute(''' curs.execute('''
CREATE TABLE 911( CREATE TABLE CO_DEUX(
latitude numeric NOT NULL, entity varchar(20) NOT NULL,
longitude numeric NOT NULL, code char(3),
description varchar(150) PRIMARY KEY, year numeric(4) NOT NULL,
zipCode numeric NOT NULL, emission numeric,
title varchar(50) NOT NULL, PRIMARY KEY(entity, year)
DateHeure timestamp NOT NULL CHECK DateHeure < CURRENT_TIMESTAMP,
ville varchar(50) NOT NULL,
adresse varchar(50) NOT NULL,
indexColone char(1) DEFAULT = 1;
); );
''') ''')
for row in df2.intertuples(): for row in df2.intertuples():
curs.execute( curs.execute(
'''INSERT INTO 911 VALUES(%f%f%s%d%s%s%s%s%d)''', '''INSERT INTO CO_DEUX VALUES(%s%s%d%f)''',
(row.Latitude, row.Longitude, row.Description, row.zipCode, row.Title, row.DateHeure, row.Ville, row.Adresse, row.indexColone) (row.entity, row.code, row.year, row.emission)
) )

Loading…
Cancel
Save