Kategorie: Programowanie

Generowanie wykresów w Pythonie

Cześć! Cieszę się, że mnie odwiedziłeś/aś. Zanim przejdziesz do artykułu chciałbym zwrocić Ci uwagę na to, że ten artykuł był pisany kilka lat temu (2013-03-31) miej więc proszę na uwadzę że rozwiązania i przemyślenia które tu znajdziesz nie muszą być aktualne. Niemniej jednak zachęcam do przeczytania.

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 – 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
Mateusz M.

Pokaż komentarze

Ostatnie wpisy

Podsumowanie: maj, czerwiec, lipiec i sierpień 2024

Oj daaawnoo mnie tu nie było. Ale wakacje to był czas dużej liczby intensywnych wyjazdów i tak naprawdę, dopiero jakoś… Read More

2 miesiące ago

Podsumowanie: kwiecień 2024

Cześć! Zapraszam na krótkie podsumowanie kwietnia. Wyjazd do Niemiec A dokładniej pod granicę z Francją. Chrześnica miała pierwszą komunię. Po… Read More

6 miesięcy ago

Podsumowanie: luty i marzec 2024

Ostatnio tygodnie były tak bardzo wypełnione, że nie udało mi się napisać nawet krótkiego podsumowanie. Więc dziś zbiorczo podsumuję luty… Read More

7 miesięcy ago

Podsumowanie: styczeń 2024

Zapraszam na krótkie podsumowanie miesiąca. Książki W styczniu przeczytałem "Homo Deus: Historia jutra". Książka łudząco podoba do wcześniejszej książki tego… Read More

9 miesięcy ago

Podsumowanie roku 2023

Cześć! Zapraszam na podsumowanie roku 2023. Książki Zacznijmy od książek. W tym roku cel 35 książek nie został osiągnięty. Niemniej… Read More

10 miesięcy ago

Podsumowanie: grudzień 2023

Zapraszam na krótkie podsumowanie miesiąca. Książki W grudniu skończyłem czytać Mein Kampf. Nudna książka. Ciekawsze fragmenty można by było streścić… Read More

11 miesięcy ago