Generowanie wykresów w Pythonie

Mateusz M. 31/03/2013

🕑 2 minuty czytania

Cześć!

Do generowania wykresów posłużę się biblioteką PyX.
Wygenerujemy serduszko :) a nawet dwa. Z tego samego wzoru, ale z inną dziedziną.

Wzór:
$x=16sin^3(x)$
$y=13cos(x)-5cos(2x)-2cos(3x)-cos(4x)$

Żeby kody które za chwile pokazę działały musicie zainstalować PyXa i LaTeXa. No i Pythona w wersji 2.7 ;)

1
2
3
4
5
6
7
8
9
10
11

import math

f=open("t.txt", "w")
s=''
for i in range(-600,600):
s=s+str(16*math.pow(math.sin(i),3))+" "+str(13*math.cos(i)-5*math.cos(2*i)-2*math.cos(3*i)-math.cos(4*i))+"\n"

print s

f.write(s)
f.close()

Nic skomplikowanego tutaj nie robimy. Tworzymy sporego stringa który będzie przechowywał w dwóch kolumnach wartości potrzebne do narysowania wykresu. Całość wrzucamy do pliku.
Pliku t.txt nie bedę wrzucał. Ma on 1200 linii… I chyba nie ma senstego robić. Można sobie go łatwo wygenerować tym programikiem wyżej.

Dobra, mamy więc punkty. Teraz skorzystajmy z PyXa i wygenerujmy wykres z tych danych:

1
2
3
4
5
6

from pyx import *

g = graph.graphxy(width=20)
g.plot(graph.data.file("t.txt", x=1, y=2), [graph.style.line([color.gradient.Rainbow])])
g.writePDFfile("heart")

Tutaj również nic odkrywczego nie ma. Wczytuje plik i koloruje wykres. Oto wygenerowany plik pdf.

Ładnie wygląda, nie? Pewna osoba nawet wolała zostać przy nim niż przy wersji poprawionej :P

Ah, wspomniałem już że bedzie wersja poprawiona, więc ta musi być błędna.. I jest błędna. Jak napisałem na początku – drugie serduszko będzie oparte na innej dziedzinie. Na tej poprawnej dziedzinie.
Spójrz proszę.. Ten przykład generuje 1200 pkt dla x <-600; 600> – sporo, nie?
Zakreś dość duży który powoduje że wykres się dubluje co daje efekt podobny do pajęczyny po bokach, bo fartem wygląda fajnie. Poprawny teraz to!

Jedyne co musimy zmieć to wartości zmiennej i które są iterowane przy tworzeniu danych. Oto poprawka:

1
2
3
4
5
6
7
8
9
10
11
12

import math

f=open("t.txt", "w")
s=''
for i in range(-600,600):
i=i*0.1
s=s+str(16*math.pow(math.sin(i),3))+" "+str(13*math.cos(i)-5*math.cos(2*i)-2*math.cos(3*i)-math.cos(4*i))+"\n"

print s

f.write(s)
f.close()

I odpalamy 2 program na nowo wygenerowanych danych co daje zupełnie inny wynik :)

Plik pdf po poprawieniu kodu.
Ładne ;) Pozdro!

Dzięki za wizytę,
Mateusz Mazurek