You've already forked etude_lego_jurassic_world
Ajoute la heatmap annuelle des personnages
This commit is contained in:
@@ -14,6 +14,11 @@ def load_counts(path: Path) -> List[dict]:
|
||||
return read_rows(path)
|
||||
|
||||
|
||||
def load_presence(path: Path) -> List[dict]:
|
||||
"""Charge le CSV de présence par année/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)
|
||||
@@ -40,3 +45,49 @@ def plot_minifigs_per_character(counts_path: Path, destination_path: Path) -> No
|
||||
fig.tight_layout()
|
||||
fig.savefig(destination_path, dpi=160)
|
||||
plt.close(fig)
|
||||
|
||||
|
||||
def plot_character_year_presence(presence_path: Path, destination_path: Path) -> None:
|
||||
"""Trace une heatmap binaire indiquant la présence d'un personnage par année."""
|
||||
rows = load_presence(presence_path)
|
||||
if not rows:
|
||||
return
|
||||
years = sorted({int(row["year"]) for row in rows})
|
||||
characters = sorted(
|
||||
{row["known_character"] for row in rows},
|
||||
key=lambda name: (
|
||||
-sum(1 for r in rows if r["known_character"] == name and r["present"] == "1"),
|
||||
name,
|
||||
),
|
||||
)
|
||||
matrix = []
|
||||
for character in characters:
|
||||
row_values = []
|
||||
for year in years:
|
||||
present = next(
|
||||
(r["present"] for r in rows if r["known_character"] == character and int(r["year"]) == year),
|
||||
"0",
|
||||
)
|
||||
row_values.append(int(present))
|
||||
matrix.append(row_values)
|
||||
|
||||
height = max(5, len(characters) * 0.35)
|
||||
fig, ax = plt.subplots(figsize=(12, height))
|
||||
cax = ax.imshow(matrix, aspect="auto", cmap="Greens", interpolation="nearest")
|
||||
ax.set_xticks(range(len(years)))
|
||||
ax.set_xticklabels(years, rotation=45, ha="right")
|
||||
ax.set_yticks(range(len(characters)))
|
||||
ax.set_yticklabels(characters)
|
||||
ax.set_xlabel("Année")
|
||||
ax.set_ylabel("Personnage")
|
||||
ax.set_title("Présence des personnages par année (hors figurants)")
|
||||
for i, character in enumerate(characters):
|
||||
for j, year in enumerate(years):
|
||||
value = matrix[i][j]
|
||||
if value == 1:
|
||||
ax.text(j, i, "●", ha="center", va="center", color="#0d0d0d", fontsize=7)
|
||||
fig.colorbar(cax, ax=ax, fraction=0.046, pad=0.04, label="Présence (1 si minifig)")
|
||||
ensure_parent_dir(destination_path)
|
||||
fig.tight_layout()
|
||||
fig.savefig(destination_path, dpi=160)
|
||||
plt.close(fig)
|
||||
|
||||
Reference in New Issue
Block a user