File size: 840 Bytes
0116d50 | 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 | """Utility plots for exploratory analysis."""
from __future__ import annotations
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
sns.set_style("whitegrid")
def plot_rating_distribution(df: pd.DataFrame):
if "overall" not in df.columns:
raise ValueError("Column 'overall' not present")
plt.figure(figsize=(7, 4))
sns.countplot(x="overall", data=df, palette="viridis")
plt.title("Ratings distribution")
return plt.gca()
def plot_cluster_sizes(labels):
series = pd.Series(labels)
counts = series.value_counts().sort_index()
plt.figure(figsize=(10, 4))
counts.plot(kind="bar", color="#0b7fab")
plt.title("Cluster sizes")
plt.xlabel("Cluster id")
plt.ylabel("# Reviews")
return plt.gca()
__all__ = ["plot_rating_distribution", "plot_cluster_sizes"]
|