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

Just-in-time compiler (JIT) w Pythonie

Cześć! W Pythonie 3.13 dodano JITa! JIT, czyli just-in-time compiler to optymalizacja na która Python naprawdę długo czekał. Na nasze… Read More

1 tydzień ago

Podsumowanie roku 2024

Cześć! Zapraszam na podsumowanie roku 2024. Książki W sumie rok 2024 był pod względem ilości książek nieco podobny do roku… Read More

2 tygodnie ago

Podsumowanie: wrzesień, październik, listopad i grudzień 2024

Podtrzymując tradycję, prawie regularnych podsumowań, zapraszam na wpis! Nie mogło obyć się bez Karkonoszy We wrześniu odwiedziłem z kolegą Karkonosze,… Read More

3 tygodnie ago

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

5 miesięcy 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

9 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

10 miesięcy ago