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:

Ż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
Podziel się na:
    Facebook email PDF Wykop Twitter

2
Dodaj komentarz

avatar
2 Wątki
0 Odpowiedzi
0 Śledzący
 
Komentarz z największą liczbą reakcji
Najczęściej komentowany wątek
2 Komentarze autora
Mateusz M.Enforcer Ostatnie komentarze autora

This site uses Akismet to reduce spam. Learn how your comment data is processed.

  Subskrybuj  
Powiadom o
Enforcer
Gość

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