Mehrere Graphen im gleichen Koordinatensystem

neuernutzer · May 23rd 2023

Hallo,

ich habe verschiedene Datensätze, die ich gerne im gleichen Koordinatensystem plotten möchte, um besser zu vergleichen.

Der Hintergrund ist der:
Ich ermittele mit einem Skript, ob eine Zahl einer bestimmten Bauform eine Primzahl ist oder nicht. Dann bekomme ich beispielsweise diese Liste dabei raus:

Code

vektor = [True, False, False, True, True, False, False, False, False, True, True]

Nun möchte ich den Verlauf plotten. Dazu ermittele ich zuerst die Anzahl der Primzahlen. Dies habe ich mit dieser Funktion realisiert:

Code

def anzahl_ermitteln(vektor):
    if vektor == False:
        return
    ywerte = []
    anzahl = 0
    for eintrag in vektor:
        if eintrag == True:
            anzahl += 1
        ywerte.append(anzahl)
    return ywerte

Dann plotte ich mit dieser Funktion:

Code

def liste_plotten(vektor):
    y_rohdaten = []
    xwerte = []
    for eintrag in vektor:
        xwerte.append(eintrag[0])
        y_rohdaten.append(eintrag[1])
    ywerte = anzahl_ermitteln(y_rohdaten)
    print("Y-Werte:", ywerte)
    plt.plot(xwerte, ywerte)
    plt.show()

Es funktioniert auch, ich bekomme einen Plot wie den Folgenden:

Nun möchte ich aber mehrere Datensätze im gleichen Koordinatensystem plotten.

Wie bekomme ich das hin?

Tell · May 23rd 2023

Ganz einfach: plot mehrfach aufrufen:

Code

import matplotlib.pyplot as plt
import numpy as np

x  = np.arange( 20 )
y1 = np.random.rand( 20 )
y2 = np.random.rand( 20 )

plt.plot( x, y1 )
plt.plot( x, y2 )

plt.show()

Display More

__blackjack__ · May 23rd 2023

neuernutzer Eine Funktion sollte in allen Zweigen einen expliziten Wert zurück, nicht mal ``return wert`` und mal nur ``return``. Wobei das erste auch gar nicht vorkommen sollte, denn an einen Namen sollten Werte vom gleichen („Duck“-)Typ gebunden werden. Also bei `vektor` eine Liste oder irgendetwas iterierbares, aber dann nicht auch noch manchmal ein Wahrheitswert (oder eine ganze Zahl).

Man macht keine Vergleiche mit literalen Wahrheitswerten. Da kommt ja nur wieder ein Wahrheitswert bei heraus. Entweder der, den man sowieso schon hatte, oder dessen Gegenteil. Dafür gäbe es dann ``not``.

Der Funktionsname ist etwas generisch. Das ist eigentlich unter „kumulative Summe“ bekannt, was da gemacht wird. Und man kann das mit `itertools.accumulate()` deutlich einfacher ausdrücken. Und zwar so einfach, dass sich da IMHO nicht wirklich eine eigene Funktion für lohnt.

Du darfst nicht direkt nach dem `plot()`-Aufruf `show()` aufrufen. Das darf erst passieren wenn der Plot komplett ist. Ich würde auch die API verwenden die auf diesem impliziten, globalen Plot operiert, sondern die objektorientierte API verwenden.

Ungetestet:

Python

#!/usr/bin/env python3
from itertools import accumulate

from matplotlib import pyplot as plt


def liste_plotten(ax, vektor):
    y_werte = list(accumulate((int(y) for _, y in vektor), initial=0))
    print("Y-Werte:", y_werte)
    ax.plot([x for x, _ in vektor], y_werte)


def main():
    ...

    figure, ax = plt.subplots()
    for vektor in vektoren:
        liste_plotten(ax, vektor)

    figure.show()


if __name__ == "__main__":
    main()

Display More

neuernutzer · May 23rd 2023

Quote from __blackjack__

[...]

Danke sehr. Das ist sehr viel was ich nicht wusste, aber darum bin ich hier. Ich werde mir das gleich mal in Ruhe anschauen. Die einzelnen Plots funktionieren jedenfalls schonmal.

Vielen Dank für eure Zeit!