可以模仿这个代码写吗def drawAxis(length): # 绘制一个给定长度的三维坐标轴 glPushAttrib(GL_POLYGON_BIT | GL_ENABLE_BIT | GL_COLOR_BUFFER_BIT) # 保存当前OpenGL状态 glPolygonMode(GL_FRONT_AND_BACK, GL_LINE) # 设置多边形绘制模式为线性模式 glDisable(GL_LIGHTING) # 关闭光照 glBegin(GL_LINES) glColor3f(1, 0, 0) glVertex3f(0, 0, 0) glVertex3f(length, 0, 0) glColor3f(0, 1, 0) glVertex3f(0, 0, 0) glVertex3f(0, length, 0) glColor3f(0, 0, 1) glVertex3f(0, 0, 0) glVertex3f(0, 0, -length) glEnd() glPopAttrib() # 恢复之前保存的OpenGL状态 print("绘制成功")

当然可以！以下是模仿上述代码编写的`drawAxis`函数：

from OpenGL.GL import *
from OpenGL.GLUT import *
from OpenGL.GLU import *

def drawAxis(length):
    glPushAttrib(GL_POLYGON_BIT | GL_ENABLE_BIT | GL_COLOR_BUFFER_BIT)  # 保存当前OpenGL状态

    glPolygonMode(GL_FRONT_AND_BACK, GL_LINE)  # 设置多边形绘制模式为线性模式
    glDisable(GL_LIGHTING)  # 关闭光照

    glBegin(GL_LINES)
    glColor3f(1, 0, 0)
    glVertex3f(0, 0, 0)
    glVertex3f(length, 0, 0)

    glColor3f(0, 1, 0)
    glVertex3f(0, 0, 0)
    glVertex3f(0, length, 0)

    glColor3f(0, 0, 1)
    glVertex3f(0, 0, 0)
    glVertex3f(0, 0, -length)
    glEnd()

    glPopAttrib()  # 恢复之前保存的OpenGL状态
    print("绘制成功")

你可以调用`drawAxis(length)`函数，并传入一个长度参数来绘制一个给定长度的三维坐标轴。注意，在使用该函数之前，你需要确保已经初始化了OpenGL环境。