python兰顿蚂蚁代码

import CV2

import numpy as np

board=np.zeros((65535,65535), np.bool_)

ant=[0,0,0]

size=1

def move():

    global size

    if board[ant[0],ant[1]]:

        ant[2]+=1

        board[ant[0],ant[1]]=False

    else:

        ant[2]-=1

        board[ant[0],ant[1]]=True

    ant[2]=ant[2]%4

    if ant[2]==0:

        ant[0]+=1

    if ant[2]==1:

        ant[1]+=1

    if ant[2]==2:

        ant[0]-=1

    if ant[2]==3:

        ant[1]-=1

    if size<abs(ant[0])+1:

        size=abs(ant[0])+1

    if size<abs(ant[1])+1:

        size=abs(ant[1])+1

while(True) :

    move()

    img=np.zeros((2*size,2*size),np.uint8)

    for x in range(-size,size):

        for y in range(-size,size):

            if board[x,y]:

                img[x+size,y+size]=255

    img=CV2.resize(img,(720,720))

    CV2.imshow('image',img) 

    CV2.waitKey(1)