4

我有一个使用 pygame 绘制 mandlebrot 集的代码。这是代码。

import pygame, sys, math
from decimal import *
window=pygame.display.set_mode((1000, 1000))
window.fill((255, 255, 255))
pygame.display.update()
winrect=window.get_rect()
hq=3
getcontext().prec=20
colors=((255, 0, 0), (255, 128, 0), (255, 255, 0), (128, 255, 0), (0, 255, 0), (0, 255, 128), (0, 255, 255), (0, 128, 255), (0, 0, 255), (128, 0, 255), (255, 0, 255), (255, 0, 128))
def graph(scale):#left, right, bottom, top
    window.fill((0, 0, 0))
    minimum=-1
    y=((scale[3]-scale[2]))/(winrect.height)+scale[2]
    for a in range(winrect.width):
        x=((scale[1]-scale[0])*(a))/(winrect.width)+scale[0]
        d, e=x**2-y**2+x, 2*x*y+y
        for i in range(int(1/(50*(scale[1]-scale[0]))+25)):
            d, e=d**2-e**2+x, 2*d*e+y
            if math.sqrt(d**2+e**2)>2:
                if i<minimum or minimum==-1:
                    minimum=i
                break
    y=((scale[3]-scale[2])*winrect.height)/(winrect.height)+scale[2]
    for a in range(winrect.width):
        x=((scale[1]-scale[0])*a)/winrect.width+scale[0]
        d, e=x**2-y**2+x, 2*x*y+y
        for i in range(int(1/(50*(scale[1]-scale[0]))+25)):
            d, e=d**2-e**2+x, 2*d*e+y
            if math.sqrt(d**2+e**2)>2:
                if i<minimum or minimum==-1:
                    minimum=i
                break
    x=((scale[1]-scale[0])*1)/winrect.width+scale[0]
    for b in range(winrect.height):
        y=((scale[3]-scale[2])*b)/winrect.height+scale[2]
        d, e=x**2-y**2+x, 2*x*y+y
        for i in range(int(1/(50*(scale[1]-scale[0]))+25)):
            d, e=d**2-e**2+x, 2*d*e+y
            if math.sqrt(d**2+e**2)>2:
                if i<minimum or minimum==-1:
                    minimum=i
                break
    x=((scale[1]-scale[0])*winrect.width)/winrect.width+scale[0]
    for b in range(winrect.height):
        y=((scale[3]-scale[2])*b)/winrect.height+scale[2]
        d, e=x**2-y**2+x, 2*x*y+y
        for i in range(int(1/(50*(scale[1]-scale[0]))+25)):
            d, e=d**2-e**2+x, 2*d*e+y
            if math.sqrt(d**2+e**2)>2:
                if i<minimum or minimum==-1:
                    minimum=i
                break
    for a in range(winrect.width):
        for b in range(winrect.height):
            x=((scale[1]-scale[0])*a)/winrect.width+scale[0]
            y=((scale[3]-scale[2])*b)/winrect.height+scale[2]
            d, e=x**2-y**2+x, 2*x*y+y
            for i in range(minimum):
                d, e=d**2-e**2+x, 2*d*e+y
            for i in range(20*hq):
                d, e=d**2-e**2+x, 2*d*e+y
                if math.sqrt(d**2+e**2)>2:
                    window.set_at((a, b), colors[i-(int(i/len(colors))*len(colors))])
                    break
            for event in pygame.event.get():
                if event.type==pygame.QUIT:
                    pygame.quit()
                    sys.exit()
                if event.type==pygame.KEYDOWN:
                    if event.key==pygame.K_ESCAPE:
                        pygame.quit()
                        sys.exit()
        pygame.display.update()
    pygame.display.update()
graph([-3, 2, -2.5, 2.5, 0])#
scale=[-3, 2, -2.5, 2.5, 0]
scalea=scale[:]
while True:
    for event in pygame.event.get():
        if event.type==pygame.QUIT:
            pygame.quit()
            sys.exit()
        if event.type==pygame.KEYDOWN:
            if event.key==pygame.K_ESCAPE:
                pygame.quit()
                sys.exit()
            if event.key==pygame.K_r:
                graph([-3, 2, -2.5, 2.5, 0])
                scale=[-3, 2, -2.5, 2.5, 0]
                scalea=scale[:]
            if event.key==pygame.K_p:
                hq+=1
                graph(scale)
            if event.key==pygame.K_o:
                if not hq==1:
                    hq-=1
                    graph(scale)
            if event.key==pygame.K_SPACE:
                print(scale)
                print(scale[1]-scale[0])
        if event.type==pygame.MOUSEBUTTONDOWN:
            if not scalea[4]:
                scalea[0]=(((scale[1]-scale[0])*event.pos[0])/winrect.width)+scale[0]
                scalea[2]=(((scale[3]-scale[2])*event.pos[1])/winrect.height)+scale[2]
                scalea[4]=1
            else:
                scalea[1]=(((scale[1]-scale[0])*event.pos[0])/winrect.width)+scale[0]
                scalea[3]=(((scale[3]-scale[2])*event.pos[1])/winrect.height)+scale[2]
                scalea[4]=0
                if scalea[1]<scalea[0]:
                    scalea=[scalea[1], scalea[0], scalea[2], scalea[3], 0]
                if scalea[3]<scalea[2]:
                    scalea=[scalea[0], scalea[1], scalea[3], scalea[2], 0]
                scale=scalea[:]
                if scale[1]-scale[0]<scale[3]-scale[2]:
                    scale[1]+=((scalea[3]-scalea[2])-(scalea[1]-scalea[0]))/2
                    scale[0]-=((scalea[3]-scalea[2])-(scalea[1]-scalea[0]))/2
                else:
                    scale[2]-=((scalea[1]-scalea[0])-(scalea[3]-scalea[2]))/2
                    scale[3]+=((scalea[1]-scalea[0])-(scalea[3]-scalea[2]))/2
                graph(scale)

当我在 python 3.2 中运行它时,它工作正常。图像如下所示:

[IMG]http://i47.tinypic.com/2ps0d8n.jpg[/IMG]

然而,当我在 python 2.7 中运行它时,我得到了一个看起来像这样的可怕图像:

[IMG]http://i47.tinypic.com/2a0nskk.jpg[/IMG]

有什么办法可以解决这个问题吗?

4

2 回答 2

10

是的,在文件顶部添加:

from __future__ import division

Python 2在默认使用整数输入时使用整数除法(地板除法);即使使用整数输入,Python 3 也切换到浮点除法。

请参阅PEP 238,其中记录了更改:

当前除法 (/) 运算符对数值参数的含义不明确:如果参数是整数或长整数,它会返回除法数学结果的下限,但如果参数是浮点数或复数,它会返回除法结果的合理近似值. 当整数不是预期的但可能作为输入时,这使得预期浮点或复杂结果的表达式容易出错。

于 2013-04-18T14:27:52.903 回答
6

from __future__ import division在顶部添加。

于 2013-04-18T14:27:44.607 回答