scatter plot in matplotlib

Question

To jest mój pierwszy program matplotlib, tak przepraszam za moją niewiedzę.

Mam dwie tablice ciągów. powiedzmy: A = ['test1','test2'] i B = ['test3','test4']. Jeśli istnieje jakaś korelacja między elementem A a B, ich wartość corr zostanie ustawiona na 1.

 test1 | test2 
test3 | 1 | 0 

test4 | 0 | 1 

Teraz chcę narysować wykres punktowy, gdzie moja oś X będzie elementy A, oś Y będzie elementy B i jeśli wartość korelacji jest 1, zostanie on oznaczony rozproszonego działki. jak to zrobić?

Answer 1

Może coś takiego:

import matplotlib.pyplot 
import pylab 

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

matplotlib.pyplot.scatter(x,y) 

matplotlib.pyplot.show() 

EDIT:

Pozwól mi zobaczyć, czy cię rozumiem poprawnie teraz:

Masz:

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

test4 | 0 | 1 | 0 

test5 | 1 | 1 | 0 

Teraz chcesz reprezentują powyższe wartości w wykresie rozproszenia, tak że wartość 1 jest reprezentowana przez kropkę.

Powiedzmy, że wyniki są przechowywane w wykazie 2-D:

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

Chcemy przekształcić je w dwóch zmiennych, więc jesteśmy w stanie je wykreślić.

I wierzę ten kod da ci to, czego szukasz:

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() 

Wskazówka że trzeba importować pylab i trzeba bawić z etykiet osi. Również to wydaje się być robotą dookoła i może być (prawdopodobnie jest) bezpośrednia metoda, aby to zrobić.