import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
from math import e, pi


fig = plt.figure()

X = np.array([re/100 + 1j*im/100 for re in range(100) for im in range(100)])
res = np.exp(X)


def plotPlochaNaRealnu():
    ax = fig.add_subplot(111, projection='3d', title='plocha na realnu')
    ax.plot_trisurf(X.real,X.imag,res.real)

    fig.show()


def plotPlochaNaImaginarnu():
    ax = fig.add_subplot(111, projection='3d', title='plocha na imaginarnu')
    ax.plot_trisurf(X.real,X.imag,res.imag)

    fig.show()