import ezdxf

# 创建一个新的DXF文档
doc = ezdxf.new(dxfversion='R2018')  # 可以指定不同的DXF版本
msp = doc.modelspace()  # 获取模型空间，用于添加图形对象

# 定义线条的起点和终点
start = (0, 0)
end = (10, 10)

# 添加线条，并设置颜色（使用颜色索引，例如红色通常是索引1）
# line = msp.add_line(start, end, dxfattribs={'color': 1})

# 或者，使用真彩色（RGB）设置颜色
# 注意：真彩色需要DXF版本R2004或更高，并且颜色值是以十六进制形式给出的（0xRRGGBB）
line = msp.add_line(start, end, dxfattribs={'color': 0xFF0000})  # 红色
line.rgb = (0, 0, 0)

# 保存DXF文件
doc.saveas("output.dxf")