1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
| import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
from matplotlib.colors import LinearSegmentedColormap
N = 100 # Refinement factor
color_list = [[i/(N-1), i/(N-1), 0] for i in range(N)]
color_list.extend(color_list[::-1])
my_cmap = mcolors.ListedColormap(color_list, name='from_list', N=None)
def fr(i):
return 1 + np.cos(np.cos(i)*5E3*np.cos(0.001*i))
x = np.linspace(-.1, .1, 10000)
y = np.linspace(-.1, .1, 10000)
X, Y = np.meshgrid(x, y)
scale = np.linspace(-1, 1., N)
plt.figure(figsize=(6,6))
F = fr(np.sqrt(X**2 + Y**2))
cs = plt.contourf(X, Y, F, aspect='equal', cmap=my_cmap)
plt.axis('off')
#plt.colorbar()
plt.show()
def yellowblack(y):
red = [(0.0, y, y), (0.45, y, y), (1.0, 0.0, 0.0)]
green = [(0.0, y, y), (0.45, y, y), (1.0, 0, 0)]
blue = [(0.0, y, 0), (0.45, y, y),(1.0,0.0,0.0)]
colordict = dict(red=red, green=green, blue=blue)
yellowblackmap = LinearSegmentedColormap('yellowblack', colordict, 256)
return yellowblackmap
plt.figure(figsize=(6,6))
F = fr(np.sqrt(X**2 + Y**2))
cs = plt.contourf(X, Y, F, aspect='equal', cmap=yellowblack(0.85))
plt.axis('off')
#plt.colorbar()
plt.show() |
Partager