-
Notifications
You must be signed in to change notification settings - Fork 0
/
Time_Series_Visualizer.py
55 lines (48 loc) · 2.23 KB
/
Time_Series_Visualizer.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
49
50
51
52
53
54
55
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from pandas.plotting import register_matplotlib_converters
register_matplotlib_converters()
# Import data (Make sure to parse dates. Consider setting index column to 'date'.)
df = pd.read_csv("fcc-forum-pageviews.csv", index_col="date" , parse_dates=True)
# Clean data
df = df[ df["value"].between( df["value"].quantile(.025), df["value"].quantile(.975) ) ]
months= ['January', 'February', 'March', 'April', 'May', 'June', 'July', 'August', 'September', 'October', 'November', 'December']
def draw_line_plot():
# Draw line plot
fig, ax = plt.subplots(figsize=(15,5))
ax = sns.lineplot(data = df, legend="brief")
ax.set(title='Daily freeCodeCamp Forum Page Views 5/2016-12/2019')
ax.set(xlabel = "Date",ylabel = "Page Views")
# Save image and return fig (don't change this part)
fig.savefig('line_plot.png')
return fig
def draw_bar_plot():
# Copy and modify data for monthly bar plot
df_bar = df.copy()
df_bar["year"] = df.index.year.values
df_bar["month"] = df.index.month_name()
# Draw bar plot
fig, ax = plt.subplots(figsize=(15,5))
ax = sns.barplot(x="year", hue="month", y="value", data=df_bar, hue_order = months, ci=None )
ax.set(xlabel = "Years",ylabel = "Average Page Views")
# Save image and return fig (don't change this part)
fig.savefig('bar_plot.png')
return fig
def draw_box_plot():
# Prepare data for box plots (this part is done!)
df_box = df.copy()
df_box.reset_index(inplace=True)
df_box['year'] = [d.year for d in df_box.date]
df_box['month'] = [d.strftime('%b') for d in df_box.date]
# Draw box plots (using Seaborn)
df_box['monthnumber'] = df.index.month
df_box = df_box.sort_values('monthnumber')
fig, ax = plt.subplots(1,2,figsize=(16,6))
sns.boxplot(y = "value", x = "year", data = df_box, ax = ax[0] )
ax[0].set(xlabel="Year", ylabel="Page Views", title="Year-wise Box Plot (Trend)")
sns.boxplot(y = "value", x = "month", data = df_box, ax = ax[1])
ax[1].set(xlabel="Month", ylabel="Page Views", title="Month-wise Box Plot (Seasonality)")
# Save image and return fig (don't change this part)
fig.savefig('box_plot.png')
return fig