python生成曼德勃罗分形图形

最新推荐文章于 2023-12-11 18:38:24 发布

墨黑白灰

最新推荐文章于 2023-12-11 18:38:24 发布

阅读量2.9k

点赞数 1

分类专栏： c++ python

本文链接：https://blog.csdn.net/Meet33/article/details/95170206

版权

c++ 同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

python

3 篇文章 0 订阅

订阅专栏

来自https://blog.csdn.net/baimafujinji/article/details/50859174

c程序版生产那种格式的图像显然不利于咱们研究分析，于是写个py版本的以方便学习。

用到了PIL的Image库来生成RGB图像，转化为numpy数组来进行像素值操作，显示用到matplotlib。

根据曼德勃罗集合迭代式，计算当前坐标是否满足条件来修改像素值，根据定理

若，则

，ok我们迭代后满足条件的像素置灰。

准备就绪，附上代码

import time
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

wid_size = 1920
hei_size = 1080
maxVal = 255
orig_x = wid_size*2/3
orig_y = hei_size/2

def iteration(x,y):
    limit = maxVal + 1
    A = (x-orig_x)/(wid_size/3)
    B = (orig_y-y)/(hei_size/2)
    c = complex(A,B)
    z = complex(0,0)
    for i in range(limit):
        z = z * z + c
        if z.real>2 or z.imag>2:
            break
    return 0 if i==maxVal else 1

if __name__ == '__main__':
    img = Image.new('RGB',(wid_size,hei_size))
    #img.shape
    imgp = np.array(img)
    t0 = time.perf_counter()
    for i in range(hei_size):
        for j in range(wid_size):
            ite = iteration(j,i)
            if ite:
                imgp[i,j,0] = ite
                imgp[i,j,1] = abs(j-orig_x)/wid_size*maxVal
                imgp[i,j,2] = abs(i-orig_y)/hei_size*maxVal
            else:
                imgp[i,j,:] = 128

    t1 = time.perf_counter()
    print(t1-t0)
    plt.imshow(imgp)
    plt.show()
    img = Image.fromarray(imgp)
    img.save('./1.jpg')

不试不知道，一试吓一跳，速度感人。。。

用了100秒，是的，效率极其低下。。。虽然写代码很方便，但还是改成c++试试吧

BYTE *imgs = new BYTE[width_size*height_size * 3];
memset(imgs, 0, width_size*height_size * 3);
double t0 = clock();
for (int i = 0; i < height_size; i++) {
	for (int j = 0; j < width_size; j++) {
		BYTE ite = iteration(j, i);
		//BGR反位写入
		if (ite) {
			imgs[(i*width_size + j) * 3 + 2] = ite;
			imgs[((i + 0)*width_size + j) * 3 + 1] = abs(j - orig_x) / width_size * Maxval;
			imgs[((i + 0)*width_size + j) * 3 + 0] = abs(i - orig_y) / height_size * Maxval;
			}
		else {
			imgs[(i*width_size + j) * 3] = 128;
			imgs[((i + 0)*width_size + j) * 3 + 1] = 128;
			imgs[((i + 0)*width_size + j) * 3 + 2] = 128;
		}
	}
}
double t1 = clock();
cout << t1 - t0 << endl;
saveBitmap(width_size, height_size, imgs, width_size * height_size * 3, "1.bmp");

好的，run一下，震惊，三秒，快了30倍，是的，嘻嘻，c++还是强的。