azalea/CSC110-Project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[O] Optimize visualization

Browse Source
This commit is contained in:
Hykilpikonna
2021-11-27 17:16:16 -05:00
parent 4b77b3e462
commit 5e5385e6af
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/process/twitter_visualization.py
+10 -11
View File
@@ -292,17 +292,17 @@ def graph_load_font() -> None:
plt.rcParams["font.family"] = "iosevka" plt.rcParams["font.family"] = "iosevka"
def graph_histogram(x: list[float], path: str, title: str, clear_outliers: bool = False, def graph_histogram(x: list[float], path: str, title: str, freq: bool, clear_outliers: bool = False,
bins: int = 20, axvline: Union[list[int], None] = None) -> None: bins: int = 20) -> None:
""" """
Plot a histogram Plot a histogram
:param x: X axis data :param x: X axis data
:param path: Output image path (should end in .png) :param path: Output image path (should end in .png)
:param title: Title :param title: Title
:param freq: Whether we are graphing frequencies data instead of popularity ratios
:param clear_outliers: Remove outliers or not :param clear_outliers: Remove outliers or not
:param bins: Number of bins :param bins: Number of bins
:param axvline: Vertical line
:return: None :return: None
""" """
if clear_outliers: if clear_outliers:
@@ -321,10 +321,10 @@ def graph_histogram(x: list[float], path: str, title: str, clear_outliers: bool
ax.set_title(title, color=border_color) ax.set_title(title, color=border_color)
ax.hist(x, bins=bins, color='#ffcccc') ax.hist(x, bins=bins, color='#ffcccc')
# Plot lines if freq:
if axvline: ax.xaxis.set_major_formatter(matplotlib.ticker.PercentFormatter(1))
for line in axvline: else:
ax.axvline(line, color='#DACAA9') ax.axvline(1, color='#DACAA9')
# Colors # Colors
ax.tick_params(color=border_color, labelcolor=border_color) ax.tick_params(color=border_color, labelcolor=border_color)
@@ -440,14 +440,13 @@ def report_histograms(sample: Sample) -> None:
""" """
x = [f.data for f in sample.user_freqs] x = [f.data for f in sample.user_freqs]
title = f'COVID-related posting frequency for {sample.name}' title = f'COVID-related posting frequency for {sample.name}'
graph_histogram(x, f'freq/{sample.name}-hist-outliers.png', title, False, 100) graph_histogram(x, f'freq/{sample.name}-hist-outliers.png', title, True, False, 100)
x = [p for p in x if p > 0.001] x = [p for p in x if p > 0.001]
graph_histogram(x, f'freq/{sample.name}-hist.png', title, True) graph_histogram(x, f'freq/{sample.name}-hist.png', title, True, True)
x = [f.data for f in sample.user_pops] x = [f.data for f in sample.user_pops]
title = f'Popularity ratio of COVID posts for {sample.name}' title = f'Popularity ratio of COVID posts for {sample.name}'
graph_histogram(x, f'pop/{sample.name}-hist-outliers.png', title, False, 100, axvline=[1]) graph_histogram(x, f'pop/{sample.name}-hist.png', title, False, True)
graph_histogram(x, f'pop/{sample.name}-hist.png', title, True, axvline=[1])
def report_stats(samples: list[Sample]) -> None: def report_stats(samples: list[Sample]) -> None: