Pasted

from rich import get_console
from rich.table import Table
 
 
def normalisiert(gesamt: list[int], anzahl_straftäter: list[int]) -> list[float]:
    """
    Berechnet für jede Bevölkerungsgruppe die normierte Straftäterrate.
 
    gesamt: Gesamtgröße der jeweiligen Bevölkerungsgruppe.
    anzahl_straftäter: Anzahl der Straftäter je Gruppe.
 
    Eine Liste mit den normierten Raten (0.0 bis 1.0).
    """
    if len(anzahl_straftäter) != len(gesamt):
        raise ValueError("Beide Listen müssen gleich lang sein.")
 
    ergebnis = []
 
    for straftäter, bevölkerung in zip(anzahl_straftäter, gesamt):
        if bevölkerung <= 0:
            raise ValueError("Alle Bevölkerungsgrößen müssen größer als 0 sein.")
        ergebnis.append(straftäter / bevölkerung)
 
    return ergebnis
 
 
def zeige_tabelle(farben: list[str], gesamt: list[int], anzahl_straftäter: list[int]):
    ergebnis = normalisiert(gesamt, anzahl_straftäter)
    summe = sum(gesamt)
    console = get_console()
    table = Table(title="Normalisierte Kriminalitätsraten")
 
    for col in (
        "Farbe",
        "Population",
        "Strafäter",
        "Rel. (gesamt)",
        "Rel. (normalisiert)",
    ):
        table.add_column(col)
 
    for farbe, anzahl, straftäter, rel_norm in zip(
        farben, gesamt, anzahl_straftäter, ergebnis
    ):
        rel_gesamt = straftäter / summe
        table.add_row(
            f"{farbe.title()} Schafe",
            f"{anzahl:>10}",
            f"{straftäter:>10}",
            f"{rel_gesamt:>7.1%}",
            f"{rel_norm:>7.1%}",
        )
 
    console.print(table)
 
 
schwarze_schafe = 90
weiße_schafe = 10
 
straftaten_schwarz = 5
straftaten_weiß = 5
 
farben = ["schwarze", "weiße"]
gesamt = [schwarze_schafe, weiße_schafe]
 
anzahl_straftäter = [straftaten_schwarz, straftaten_weiß]
zeige_tabelle(farben, gesamt, anzahl_straftäter)