You've already forked etude_lego_jurassic_world
Ajoute le graphique des personnages représentés
This commit is contained in:
42
lib/plots/minifig_characters.py
Normal file
42
lib/plots/minifig_characters.py
Normal file
@@ -0,0 +1,42 @@
|
||||
"""Graphique du nombre de minifigs par personnage."""
|
||||
|
||||
from pathlib import Path
|
||||
from typing import List
|
||||
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
from lib.filesystem import ensure_parent_dir
|
||||
from lib.rebrickable.stats import read_rows
|
||||
|
||||
|
||||
def load_counts(path: Path) -> List[dict]:
|
||||
"""Charge le CSV des comptes par personnage."""
|
||||
return read_rows(path)
|
||||
|
||||
|
||||
def plot_minifigs_per_character(counts_path: Path, destination_path: Path) -> None:
|
||||
"""Trace un diagramme en barres horizontales du nombre de minifigs par personnage."""
|
||||
rows = load_counts(counts_path)
|
||||
characters = [row["known_character"] for row in rows]
|
||||
counts = [int(row["minifig_count"]) for row in rows]
|
||||
positions = list(range(len(rows)))
|
||||
height = max(6, len(rows) * 0.22)
|
||||
|
||||
fig, ax = plt.subplots(figsize=(12, height))
|
||||
bars = ax.barh(positions, counts, color="#1f77b4", edgecolor="#0d0d0d", linewidth=0.6)
|
||||
ax.set_yticks(positions)
|
||||
ax.set_yticklabels(characters)
|
||||
ax.invert_yaxis()
|
||||
ax.set_xlabel("Nombre de minifigs distinctes")
|
||||
ax.set_title("Minifigs par personnage (thèmes filtrés)")
|
||||
ax.grid(True, axis="x", linestyle="--", alpha=0.25)
|
||||
max_value = max(counts) if counts else 0
|
||||
ax.set_xlim(0, max_value + 1)
|
||||
for index, bar in enumerate(bars):
|
||||
value = counts[index]
|
||||
ax.text(value + 0.1, bar.get_y() + bar.get_height() / 2, str(value), va="center", fontsize=8)
|
||||
|
||||
ensure_parent_dir(destination_path)
|
||||
fig.tight_layout()
|
||||
fig.savefig(destination_path, dpi=160)
|
||||
plt.close(fig)
|
||||
Reference in New Issue
Block a user