Python数列绘图

前言

笔者最近两天接触了Python语言，在学校刚学完了高中必修5的数列部分后，便萌生了用计算机计算数列的想法，顺便实现了数列的绘图（turtle库实现，待完善）。

效果图

说明

需输入的delay为输出结果的延迟时间，单位毫秒，输入的表达式可以使用random库，math库，time库中所有函数，但要求符合Python语法。若要计算数列b(n)的前n项和，则可把表达式写成a(n) = a(n-1) + b(n)，此时a(1) = b(1)

代码

import turtle
from random import *
from math import *
from time import *

n = 1
a = 0

def printa(a):
    global n
    turtle.goto((n % 1280) - 640,a * 1)
    print("a(" + str(n) + ") = " + str(a))
    n += 1

def format(s):
    s = s.replace("^","**")
    return s

delay = int(eval(input("delay = ")) * 1e-3)
express = format(input("a(n) = "))

turtle.setup(1280, 720)
turtle.penup()
turtle.goto(-640,0)
turtle.pendown()

if "a(n-1)" in express:
    express = express.replace("a(n-1)","a")
    a = eval(input("a(1) = "))
    n = 2

while  True:
    a = eval(express)
    printa(a)
    sleep(delay)