Python — vánoční stromek

Jaké by to byly Vánoce bez vánočního stromku! Je jasné, že každý správný Pythonista musí umět takový stromek naprogramovat. Nejlépe na pár řádcích.

Pokud to Pythonista neumí nebo se chce inspirovat, přichází ke slovu internet. Ten je sice plný návodů, ale s jejich kvalitou je to někdy horší. My jsme si dali tu práci a celý internet prohledali, takže vy už nemusíte.

Kód jsme pročistili a opatřili poznámkami, abyste si s ním mohli snadno hrát a upravovat ho. Stačí jen zkopírovat. Na 1 kliknutí!

Stromek využívá knihovny numpy a matplotlib. Pokud používáte distribuci Anaconda, měli byste mít obě knihovny už nainstalovány.

Přejeme vám Lovely Vánoce!

# Import knihoven
import numpy as np
import matplotlib.pyplot as plt

# Funkce pro výpočet souřadnic spirály
def tree_coordinates():
    """Vypočítá souřadnice překrásného stromu"""
    theta = np.linspace(-10 * np.pi, 10 * np.pi, 250) 
    z = np.linspace(-10, 0, 250)
    x = np.sin(theta) * z
    y = np.cos(theta) * z
    return (x,y,z)

fig = plt.figure(figsize=(10, 10))  # Velikost obrázku v palcích
ax = fig.gca(projection='3d')       # Chceme stromek ve 3D
ax.view_init(10, 5)                 # Nastavíme úhel pohledu
ax._axis3don = False                # Skryjeme osy
 
x, y, z = tree_coordinates()        # Načteme souřadnice

ax.plot(x, y, z,                    # Strom
        c='#900090', linewidth=5)
    
ax.scatter(0, 0, 1,                 # Hvězda
           c='#FFCE0A', s=1000, 
           marker='*')

plt.show()                          # Zobrazíme výsledek