怎么在Python中利用matplotlib绘制矩形

这篇文章给大家介绍怎么在Python中利用matplotlib绘制矩形，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。

假设矩形两点坐标如下，分别为：x1, y1, x2, y2

cat_dict['bbox'][i]=(min_row,min_col,max_row,max_col)

1. plt.plot(x,y)

这种方式画的矩形 因为边距的问题 会放缩

plt.plot([cat_dict['bbox'][i][1],cat_dict['bbox'][i][3],cat_dict['bbox'][i][3],cat_dict['bbox'][i][1],cat_dict['bbox'][i][1]],#col
[cat_dict['bbox'][i][0],cat_dict['bbox'][i][0],cat_dict['bbox'][i][2],cat_dict['bbox'][i][2],cat_dict['bbox'][i][0]],#row
color=[c/255forcinlabel_colors[cat_idx]],marker='.',ms=0)

2. ax.add_line(Line2D)

添加 4 条直线的方式，比较繁琐

frommatplotlib.linesimportLine2D
color=[c/255forcinlabel_colors[cat_idx]]
ax.add_line(Line2D([cat_dict['bbox'][i][1],cat_dict['bbox'][i][3]],
[cat_dict['bbox'][i][0],cat_dict['bbox'][i][0]],
linewidth=2,
color=color))
ax.add_line(Line2D([cat_dict['bbox'][i][3],cat_dict['bbox'][i][3]],
[cat_dict['bbox'][i][0],cat_dict['bbox'][i][2]],
linewidth=2,
color=color))
ax.add_line(Line2D([cat_dict['bbox'][i][3],cat_dict['bbox'][i][1]],
[cat_dict['bbox'][i][2],cat_dict['bbox'][i][2]],
linewidth=2,
color=color))
ax.add_line(Line2D([cat_dict['bbox'][i][1],cat_dict['bbox'][i][1]],
[cat_dict['bbox'][i][2],cat_dict['bbox'][i][0]],
linewidth=2,
color=color))

3. plt.gca().add_patch(plt.Rectangle())

最好的一种实现方式，fast rcnn 也是这么用的，传送门

plt.gca().add_patch(plt.Rectangle(xy=(cat_dict['bbox'][i][1],cat_dict['bbox'][i][0]),
width=cat_dict['bbox'][i][3]-cat_dict['bbox'][i][1],
height=cat_dict['bbox'][i][2]-cat_dict['bbox'][i][0],
edgecolor=[c/255forcinlabel_colors[cat_idx]],
fill=False,linewidth=2))

关于怎么在Python中利用matplotlib绘制矩形就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。