Python+numbaでマンデルブロ集合を描く（今更）、、、colormapsの色を確かめるために！

最近、家にいることが多いのでプログラムでも組んでみようと思った。Pythonなら会社でも使えるしまあいいかなということで

お絵かきから、、、で早速カラーマップで悩む、、、どれがいいんだ、、、

ということでここは実際にお絵かき、Mandelbrot集合あたりがいいだろうということでやってみた。

さすがにnumbaのjitで計算は一瞬です。

Spectral

Spectral_r

amhot

amhot_r

twilight_shifted

coolwarm_r

YlOrRd_r

うーん、afmhotあたりか、、、

プログラムはこちら。cm.????のところを変えれば色が変わる。

import numpy as np
import matplotlib.pyplot as plt
from numba import jit

@jit
def mandel_calc(imag, Nx, Ny, xmin, xmax, ymin, ymax, max_iteration):

    x=np.linspace(xmin,xmax,Nx)
    y=np.linspace(ymin,ymax,Ny)

    for i in range(Nx):
        for j in range(Ny):
            iteration=0
            c=x[i]+1j*y[j]
            z=0+0j
            while z.real**2+z.imag**2<4 and iteration < max_iteration:
                z=z**2+c
                iteration=iteration+1
           imag[j][i]=np.log(iteration)

Nx=1200
Ny=1200
max_iteration=1000
imag=np.zeros((Nx,Ny))
xmax=1.0
xmin=-2.5
ymax=1.5
ymin=-1.5
mandel_calc(imag, Nx, Ny, xmin, xmax, ymin, ymax,max_iteration)
plt.figure(figsize=(12,12))
plt.imshow(imag,extent=(xmin,xmax,ymin,ymax),cmap=plt.cm.coolwarm_r)
# plt.colorbar()
plt.show()