Mateusz Mazurek – programista z pasją

Blog o Pythonie i kilku innych technologiach. Od developera dla wszystkich.

Programowanie

Generowanie wykresów w Pythonie

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

A może wolisz nowości na mail?

Subskrybuj
Powiadom o
guest

Witryna wykorzystuje Akismet, aby ograniczyć spam. Dowiedz się więcej jak przetwarzane są dane komentarzy.

2 komentarzy
Inline Feedbacks
View all comments
Enforcer

Ciekawe, czy poradzi sobie z równaniem Batmana…
http://i.stack.imgur.com/VYKfg.jpg