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

Loading…
Cancel
Save