python绘制禁掉区范围

import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle

# 读取数据
data=pd.read_csv(r'D:\temp\工作簿2.csv',encoding='utf-8')

prohibitedCount=1 # 设置禁吊区数量，默认data最后的几行为禁掉区

# 创建图形和轴
fig, ax = plt.subplots()
x=[]

# 遍历每行数据并绘制矩形
for index, row in data.iterrows():
    # 提取左下和右上坐标
    left_bottom = (row['addr_x_fr'], row['addr_y_fr'])
    right_top = (row['addr_x_to'], row['addr_y_to'])
    x.append(index)
    # 计算矩形的宽度和高度
    width = right_top[0] - left_bottom[0]
    height = right_top[1] - left_bottom[1]
    
    # 创建矩形对象
    if index<=len(data)-prohibitedCount-1:
        rect = Rectangle(left_bottom, width, height, linewidth=1, edgecolor='blue', facecolor='none')
    else:
        rect = Rectangle(left_bottom, width, height, linewidth=1, edgecolor='r', facecolor='none')
    
    # 将矩形添加到轴上
    ax.add_patch(rect)

# 设置轴的范围
# 设置轴的范围
ax.set_xlim(0, 320000)
ax.set_ylim(0, 36000)
ax.set_aspect('equal')


plt.savefig('rectangles_2.svg', format='svg')  # 保存为SVG格式的矢量图


# 显示图形
plt.show()