-
Notifications
You must be signed in to change notification settings - Fork 0
/
my_functions.py
48 lines (37 loc) · 1.09 KB
/
my_functions.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
def load_dataset():
import os
import pandas as pd
from dotenv import load_dotenv
from sqlalchemy import create_engine
load_dotenv()
DB_USER = os.environ["DB_USER"]
DB_PASSWORD = os.environ["DB_PASSWORD"]
DB_HOSTNAME = os.environ["DB_HOSTNAME"]
DB_NAME = os.environ["DB_NAME"]
print(f"If everything is set up correctly this will print the database name: {DB_NAME}")
engine = create_engine(f"postgresql://{DB_USER}:{DB_PASSWORD}@{DB_HOSTNAME}:5432/{DB_NAME}")
result = engine.execute("SELECT 1")
result.first()
# regendata
df1 = pd.read_sql_query(
"""
select *
from
covid."Rain_Data"
""",
con=engine,
)
# Orderdata
df2 = pd.read_sql_query(
"""
select *
from covid."Orders_Rotterdam" om
""",
con=engine,
)
df1.rename(columns={"DATE": "date"}, inplace=True)
df1["date"] = pd.to_datetime(df1["date"])
df2["date"] = pd.to_datetime(df2["date"])
df = pd.merge(df1, df2, on="date")
df.drop(["index_x", "index_y"], axis=1, inplace=True)
return df