Skip to content
/ plotcp Public

Python package for drawing transformations of functions of a complex variable of the whole grid or a given area

License

Notifications You must be signed in to change notification settings

ZettZet/plotcp

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

69 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Plotly version

Plotly version can be found here

PlotComplexPlane

Python library for plotting complex functions transformations

It can...

  • plot complex planes (both transformed and original)
  • plot transformations of specific areas (both transformed and original)
  • plot lines parallel to real or imaginary axis (both transformed and original)

Usage

plotcp_plt

To plot f(z) = (z+1)/z with x bound from -4 to 4 and y bound from -4 to 4

from plotcp import plotcp


def f(z: complex) -> complex:  # Define function to plot
    return (z + 1) / z


# Call plotcp_plt
# Second and third arguments define limits of a plot
ax = plotcp(f, (-4, 4), (-4, 4))

For full parameters list check help(plotcp_plt.plotcp_plt)

plot_complex_points_plt

from cmath import sin

import matplotlib.pyplot as plt
import numpy as np

from plotcp import plot_complex_points


def f(z: complex) -> complex:  # Define function to plot
    return sin(z)


# Define area to be plotted
top = [x + 2 * 1j for x in np.linspace(1, 2, 5)]
bottom = [x + 1 * 1j for x in np.linspace(1, 2, 5)]
left = [1 + y * 1j for y in np.linspace(1, 2, 5)]
right = [2 + y * 1j for y in np.linspace(1, 2, 5)]

# Plot original area
ax = plot_complex_points(top)
ax = plot_complex_points(bottom, ax=ax)
ax = plot_complex_points(left, ax=ax)
ax = plot_complex_points(right, ax=ax)

# Apply function to area and plot it on a new plot
ax2 = plot_complex_points([f(z) for z in top])
ax2 = plot_complex_points([f(z) for z in bottom], ax=ax2)
ax2 = plot_complex_points([f(z) for z in left], ax=ax2)
ax2 = plot_complex_points([f(z) for z in right], ax=ax2)

plt.show()

For full parameters list check help(plotcp.plot_complex_points)

About

Python package for drawing transformations of functions of a complex variable of the whole grid or a given area

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published