In [2]:

    

import math
from matplotlib import cm as cm
from matplotlib import pyplot as plt
import numpy as np

xs = np.linspace(-3.14, 3.14, num=200)
ys = np.linspace(-1, 1, num=200)

zs = [ [x + 1.0j*y for x in xs] for y in ys]
zs = np.array(zs)

image = np.abs(np.sin(zs))


    

In [3]:

    

im = plt.imshow(image, aspect='auto')
plt.xlabel('x')
plt.ylabel('y')
plt.title('plot of abs(sin(x+i*y))')
plt.show()


    

In [18]:

    

# http://matplotlib.org/examples/color/colormaps_reference.html
cmaps = [ None, 'viridis', 'inferno', 'plasma', 'Blues', 'afmhot', 'RdBu']
cmaps = [ cm.get_cmap(name) for name in cmaps]

for cmap in cmaps:
    plt.imshow(image, aspect='auto', cmap=cmap)
    plt.show()