diff --git a/README.md b/README.md index c681a83..38b156c 100644 --- a/README.md +++ b/README.md @@ -274,7 +274,7 @@ Le script lit `data/intermediate/minifigs_by_set.csv` et `data/intermediate/sets 1. `source .venv/bin/activate` 2. `python -m scripts.plot_minifig_gender_share` -Le script lit `data/intermediate/minifigs_by_set.csv`, agrège le nombre de minifigs distinctes par genre (basé sur `config/known_character_genders.csv`), écrit `data/intermediate/minifig_gender_counts.csv`, puis trace `figures/step25/minifig_gender_share.png` (donut indiquant la part des personnages féminins, masculins ou inconnus). +Le script lit `data/intermediate/minifigs_by_set.csv`, agrège le nombre de minifigs distinctes par genre (basé sur `config/known_character_genders.csv`), écrit `data/intermediate/minifig_gender_counts.csv`, puis trace `figures/step25/minifig_gender_share.png` (donut indiquant la part des personnages féminins et masculins, les genres inconnus étant ignorés pour ce graphique). ### Étape 26 : corrélation pièces / minifigs diff --git a/lib/plots/minifig_gender_share.py b/lib/plots/minifig_gender_share.py index 20c2d92..f3c537b 100644 --- a/lib/plots/minifig_gender_share.py +++ b/lib/plots/minifig_gender_share.py @@ -17,7 +17,9 @@ def load_gender_counts(path: Path) -> List[dict]: def plot_minifig_gender_share(counts_path: Path, destination_path: Path) -> None: """Trace un diagramme circulaire de la répartition des minifigs par genre.""" - rows = load_gender_counts(counts_path) + rows = [ + row for row in load_gender_counts(counts_path) if row["gender"].strip().lower() in ("male", "female") + ] if not rows: return genders = [row["gender"] for row in rows] diff --git a/tests/test_minifig_gender_share_plot.py b/tests/test_minifig_gender_share_plot.py index 2f37f23..8b28123 100644 --- a/tests/test_minifig_gender_share_plot.py +++ b/tests/test_minifig_gender_share_plot.py @@ -17,7 +17,7 @@ def test_plot_minifig_gender_share(tmp_path: Path) -> None: "gender,minifig_count\n" "male,2\n" "female,1\n" - "unknown,1\n" + "unknown,5\n" ) plot_minifig_gender_share(counts_path, destination)