pygame写分形几何

曼德博集合（Mandelbrot set，或译为曼德布洛特复数集合）是一种在复平面上组成分形的点的集合，以数学家本华·曼德博的名字命名。曼德博集合与朱利亚集合有些相似的地方，例如使用相同的复二次多项式来进行迭代。  ---维基百科

import pygame,sys,time

from pygame.locals import *



FPS = 30

WINDOWWIDTH = 640

WINDOWHEIGHT = 480



MAX_WIDTH = 1.0

MIN_WIDTH = -2.0

MAX_HEIGHT = 1.0

MIN_HEIGHT = -1.0







BLUE =  (0,152,0)

BLACK = (10,2,2)



def generateMandelbrot():

	mandel = pygame.Surface((WINDOWWIDTH,WINDOWHEIGHT))

	mandel.fill(BLACK)



	pixObj = pygame.PixelArray(mandel)

	

	for h in range(0,WINDOWHEIGHT):

		for w in range(0,WINDOWWIDTH):

			x = MIN_WIDTH + (MAX_WIDTH - MIN_WIDTH)*w / WINDOWWIDTH

			y = MIN_HEIGHT+ (MAX_HEIGHT - MIN_HEIGHT)*h / WINDOWHEIGHT

			

			c = x + y*1j

			z = 0

			for i in range(1,100):

				z = z*z + c

				if abs(z) > 10.0 :

					break

			if i >= 99:

				pixObj[w][h] = BLUE			





	del pixObj



	return mandel



def main():

	global DISPLAYSURF

	pygame.init()

	FPSCLOCK = pygame.time.Clock()

	DISPLAYSURF = pygame.display.set_mode((WINDOWWIDTH,WINDOWHEIGHT))

		

	mandel = generateMandelbrot()

	DISPLAYSURF.blit(mandel,(0,0))



	while True:		

		for event in pygame.event.get(QUIT):

			pygame.quit()

			sys.exit()

		pygame.display.update()

		FPSCLOCK.tick(FPS)



main()

结果如下