Streudiagramm in Matplotlib

Dies ist meine erste matplotlib Programm, also sorry für meine Unwissenheit.

Ich habe zwei arrays von Strings. sagen, A = ['test1','test2'] und B = ['test3','test4'].
Wenn eine Korrelation besteht zwischen A und B element, Ihre corr-Wert wird festgelegt, um 1.

        test1 | test2
test3 |   1   |   0

test4 |   0   |   1

Nun, ich will zeichnen Sie ein Streudiagramm, wo mein X-Achse werden die Elemente der A an der Y-Achse werden die Elemente der B und wenn der korrelationswert ist 1 es gekennzeichnet werden müssen, in der verstreut Grundstück. wie macht man das?

InformationsquelleAutor der Frage bray | 2012-04-26

  1. 68

    Vielleicht so etwas wie dieses:

    import matplotlib.pyplot
import pylab

x = [1,2,3,4]
y = [3,4,8,6]

matplotlib.pyplot.scatter(x,y)

matplotlib.pyplot.show()

    EDIT:

    Lassen Sie mich sehen, wenn ich Sie richtig verstehe, jetzt:

    Haben Sie:

           test1 | test2 | test3
test3 |   1   |   0  |  1

test4 |   0   |   1  |  0

test5 |   1   |   1  |  0

    Nun wollen Sie repräsentieren die oben genannten Werte in ein Streudiagramm, so dass der Wert 1 ist dargestellt durch einen Punkt.

    Let ' s sagen Sie Ihnen Ergebnisse werden in einer 2-D Liste:

    results = [[1, 0, 1], [0, 1, 0], [1, 1, 0]]

    Wir wollen, verwandeln Sie sich in zwei Variablen, so sind wir in der Lage zu zeichnen.

    Und ich glaube, dieser code wird Euch geben, was Sie suchen:

    import matplotlib
import pylab


results = [[1, 0, 1], [0, 1, 0], [1, 1, 0]]

x = []
y = []

for ind_1, sublist in enumerate(results):
    for ind_2, ele in enumerate(sublist):
        if ele == 1:
            x.append(ind_1)
            y.append(ind_2)       


matplotlib.pyplot.scatter(x,y)

matplotlib.pyplot.show()

    Beachten Sie, dass ich tun müssen, importieren pylabund Sie würden spielen, um mit den Achsenbeschriftungen. Auch das fühlt sich an wie eine Arbeit um, und es gibt vielleicht (wahrscheinlich) eine direkte Methode, dies zu tun.

    InformationsquelleAutor der Antwort Akavall

Schreibe einen Kommentar