Skip to content
This repository has been archived by the owner on Feb 14, 2024. It is now read-only.

Commit

Permalink
Merge d27e273 into 31f8630
Browse files Browse the repository at this point in the history
  • Loading branch information
@DJensen94
DJensen94 committed Feb 10, 2023
2 parents 31f8630 + d27e273 commit 069110f
Showing 1 changed file with 118 additions and 45 deletions.
163 changes: 118 additions & 45 deletions src/pe_reports/charts.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,13 @@
import os

# Third-Party Libraries
import matplotlib
import matplotlib.pyplot as plt
from matplotlib.ticker import MaxNLocator

matplotlib.use("Agg")


# Factor to convert cm to inches
CM_CONVERSION_FACTOR = 2.54

Expand Down Expand Up @@ -49,16 +53,16 @@ def autopct(pct):
pie = plt.pie(
value_column,
startangle=0,
radius=1,
radius=1.75,
autopct=autopct,
textprops={"color": "w", "fontsize": 7},
)
plt.legend(
pie[0],
labels,
bbox_to_anchor=(1, 0.5),
bbox_to_anchor=(1, 0.75),
loc="center right",
fontsize=7,
fontsize=6,
bbox_transform=plt.gcf().transFigure,
frameon=False,
)
Expand All @@ -82,22 +86,18 @@ def stacked_bar(self):
name = self.name
color = ["#1357BE", "#D0342C"]
df.plot(kind="bar", stacked=True, zorder=3, color=color)
# Add title to chart
plt.title(title, pad=15, fontsize=10)
# Format chart's axis
plt.xlabel(x_label, labelpad=10, fontdict={"size": 8})
plt.ylabel(y_label, labelpad=10, fontdict={"size": 8})
plt.gca().yaxis.set_major_locator(MaxNLocator(integer=True))
plt.rc("axes", axisbelow=True)
plt.grid(axis="y", zorder=0)
plt.xticks(rotation=0)
plt.ylim(ymin=0)
# Set sizing for image
plt.gcf().set_size_inches(
width / CM_CONVERSION_FACTOR, height / CM_CONVERSION_FACTOR
)
plt.ylim(ymin=0)
plt.tight_layout()
# Save chart to assets directory
plt.gca().yaxis.set_major_locator(MaxNLocator(integer=True))
plt.rc("axes", axisbelow=True)
plt.grid(axis="y", zorder=0)
plt.xticks(rotation=0)
plt.savefig(BASE_DIR + "/assets/" + name, transparent=True, dpi=500)
plt.clf()

Expand All @@ -116,11 +116,10 @@ def h_bar(self):
value_column = df[df.columns[1]]
bar_width = 0.6
fig, ax = plt.subplots()
ax.spines.right.set_visible(False)
ax.spines.top.set_visible(False)
# Generate horizontal bar chart
plt.set_loglevel("WARNING")
ax.spines["right"].set_visible(False)
ax.spines["top"].set_visible(False)
plt.barh(df.index, value_column, bar_width, align="center", color="#466fc6")
# Specify axis atributes
plt.xticks(fontsize=7)
plt.yticks(fontsize=7)
plt.xlim(xmin=0)
Expand All @@ -129,13 +128,12 @@ def h_bar(self):
plt.gca().set_yticks(df.index)
plt.gca().set_yticklabels(category_column)
plt.gca().set_xlabel(x_label, fontdict={"size": 8})
plt.gca().set_ylabel(y_label)
# Set sizing for image
plt.gca().set_ylabel(y_label, fontdict={"size": 8})
plt.gcf().set_size_inches(
width / CM_CONVERSION_FACTOR, height / CM_CONVERSION_FACTOR
)
plt.tight_layout()
# Add data labels to each bar if greater than 0

for i in range(len(df)):
if df.loc[i, value_name] > 0:
label = df.loc[i, value_name]
Expand All @@ -147,7 +145,6 @@ def h_bar(self):
ha="center", # horizontal alignment can be left, right or center
fontsize=8,
)
# Save chart to assets directory
plt.savefig(
BASE_DIR + "/assets/" + name, transparent=True, dpi=500, bbox_inches="tight"
)
Expand All @@ -161,44 +158,120 @@ def line_chart(self):
width = self.width
height = self.height
name = self.name
color = ["#1357BE", "#D0342C"]
value_column = df[df.columns[0]]
color = ["#7aa5c1", "#e08493"]
fig, ax = plt.subplots()
ax.spines["right"].set_visible(False)
ax.spines["top"].set_visible(False)
plt.set_loglevel("WARNING")
# Generate lines for chart and add data labels
for col in range(len(df.columns)):
plt.plot(
df.index,
value_column,
color=color[0],
label=df.columns[0],
linewidth=3,
marker=".",
markersize=10,
)
if len(df.columns) == 2:
plt.plot(
df.index, df[df.columns[col]], color=color[col], label=df.columns[col]
df.index,
df[df.columns[1]],
color=color[1],
label=df.columns[1],
linewidth=3,
linestyle="dashed",
marker=".",
markersize=10,
)
for i, j in df[df.columns[col]].items():
if int(j):
plt.annotate(
str(int(j)),
xy=(i, j),
textcoords="offset points", # how to position the text
xytext=(
0,
5,
), # distance from text to points (x,y)
ha="center", # horizontal alignment can be left, right or center
)
# Specify axis attributes
plt.ylim(ymin=0, ymax=int(df[df.columns].max().max() * 1.15))
y_max = int(df[df.columns].max().max() * 1.1)
plt.ylim(ymin=0, ymax=y_max)
# plt.legend(loc=9, ncol=2, framealpha=0, fontsize=8, bbox_to_anchor=(0.5, -0.5))
plt.legend(loc="upper right")
plt.gcf().set_size_inches(
width / CM_CONVERSION_FACTOR, height / CM_CONVERSION_FACTOR
)
plt.xticks(fontsize=7)
plt.yticks(fontsize=7)
plt.gca().set_ylabel(y_label, labelpad=10, fontdict={"size": 8})
plt.xlabel(x_label, labelpad=10, fontdict={"size": 8})
plt.xticks(rotation=0)
plt.grid(axis="y")
# Add legend
plt.legend(loc="upper right")
# Set sizing for image
plt.gcf().set_size_inches(
width / CM_CONVERSION_FACTOR, height / CM_CONVERSION_FACTOR
)
plt.tight_layout()
# Save chart to assets directory

# loop through the dataframe
for row in df.itertuples():
# check if there is only one row of values
if len(row) == 2:
plt.annotate(
str(int(row[1])),
xy=(row[0], row[1]),
textcoords="offset points", # how to position the text
xytext=(
0,
8,
), # distance from text to points (x,y)
ha="center", # horizontal alignment can be left, right or center
# fontsize=2,
color="#003e67",
)
# check if there are two rows of data
elif len(row) == 3:
# check if the two values are within 1/10th of the max y value
value_diff = abs(row[1] - row[2])
if value_diff < y_max / 10:
# if the values are on the bottom quarter of the graph don't label below values
if min(row[1], row[2]) < y_max / 4:
y1 = y2 = max(row[1], row[2])
if row[1] > row[2]:
y1_offset = 18
y2_offset = 8
else:
y1_offset = 8
y2_offset = 18
else:
y1 = row[1]
y2 = row[2]
if row[1] > row[2]:
y1_offset = 8
y2_offset = -17
else:
y1_offset = -17
y2_offset = 8
# if values are not close to each other put the labels directly above
else:
y1 = row[1]
y2 = row[2]
y1_offset = 8
y2_offset = 8
print(y1)
print(y2)

plt.annotate(
str(int(row[1])),
xy=(row[0], y1),
textcoords="offset points", # how to position the text
xytext=(
0,
y1_offset,
), # distance from text to points (x,y)
ha="center", # horizontal alignment can be left, right or center
# fontsize=2,
color="#005288",
)
plt.annotate(
str(int(row[2])),
xy=(row[0], y2),
textcoords="offset points", # how to position the text
xytext=(
0,
y2_offset,
), # distance from text to points (x,y)
ha="center", # horizontal alignment can be left, right or center
# fontsize=2,
color="#c41230",
)

plt.savefig(
BASE_DIR + "/assets/" + name, transparent=True, dpi=500, bbox_inches="tight"
)
Expand Down

0 comments on commit 069110f

Please sign in to comment.