拉伸矩形：
import pandas as pd
import matplotlib.patches as mpathes
from matplotlib import pyplot as plt
def stretch_bb(bb, shift_x, shift_y, free_points=None):
"""
p2----------p3
| |
| |
p1----------p4
:param bb:
:param shift_x:
:param shift_y:
:param free_points:
:return:
"""
minx, miny, maxx, maxy = bb
p1 = (minx, miny)
p2 = (minx, maxy)
p3 = (maxx, maxy)
p4 = (maxx, miny)
new_point_lst = []
for free, point in zip(free_points, (p1, p2, p3, p4)):
if free:
point = point[0] + shift_x, point[1] + shift_y
new_point_lst.append(point)
minx = min([i[0] for i in new_point_lst])
miny = min([i[1] for i in new_point_lst])
maxx = max([i[0] for i in new_point_lst])
maxy = max([i[1] for i in new_point_lst])
return minx, miny, maxx, maxy
def stretch_data_lst(data_lst, base_point, shift_lst):
shift_l, shift_t, shift_r, shift_b = shift_lst
BaseX = base_point[0]
BaseY = base_point[1]
new_bb_lst = []
for data in data_lst:
x0, y0, x1, y1 = data
if shift_l:
free0 = free1 = free2 = free3 = False
if x0 < BaseX:
free0 = free1 = True
if x1 < BaseX:
free2 = free3 = True
free_points = free0, free1, free2, free3
x0, y0, x1, y1 = stretch_bb(bb=(x0, y0, x1, y1), shift_x=shift_l, shift_y=0, free_points=free_points)
if shift_t:
free0 = free1 = free2 = free3 = False
if y1 > BaseY:
free1 = free2 = True
if y0 > BaseY:
free0 = free3 = True
free_points = free0, free1, free2, free3
x0, y0, x1, y1 = stretch_bb(bb=(x0, y0, x1, y1), shift_x=0, shift_y=shift_t, free_points=free_points)
if shift_r:
free0 = free1 = free2 = free3 = False
if x0 > BaseX:
free0 = free1 = True
if x1 > BaseX:
free2 = free3 = True
free_points = free0, free1, free2, free3
x0, y0, x1, y1 = stretch_bb(bb=(x0, y0, x1, y1), shift_x=shift_r, shift_y=0, free_points=free_points)
if shift_b:
free0 = free1 = free2 = free3 = False
if y0 < BaseY:
free0 = free3 = True
if y1 < BaseY:
free1 = free2 = True
free_points = free0, free1, free2, free3
x0, y0, x1, y1 = stretch_bb(bb=(x0, y0, x1, y1), shift_x=0, shift_y=shift_b, free_points=free_points)
new_bb_lst.append((x0, y0, x1, y1))
return new_bb_lst
def get_bb_lst_center(bb_lst:list):
minx, miny, maxx, maxy = float('inf'), float('inf'), float('-inf'), float('-inf')
for bb in bb_lst:
x0, y0, x1, y1 = bb
minx = min(minx, x0)
miny = min(miny, y0)
maxx = max(maxx, x1)
maxy = max(maxy, y1)
return (minx+maxx)/2, (miny+maxy)/2
bb_lst = [(102660, 52530, 122080, 53530), (102660, 52530, 103660, 76040), (121080, 52530, 122080, 76040), (102660, 75040, 122080, 76040), (102660, 52530, 122080, 53530), (102660, 52530, 103660, 76040), (121080, 52530, 122080, 76040), (102660, 75040, 122080, 76040), (121470, 53835, 121690, 54055), (121470, 54305, 121690, 54525), (121470, 54775, 121690, 54995), (121470, 55245, 121690, 55465), (121470, 55715, 121690, 55935), (121470, 56185, 121690, 56405), (121470, 56655, 121690, 56875), (121470, 57125, 121690, 57345), (121470, 57595, 121690, 57815), (121470, 58065, 121690, 58285), (121470, 58535, 121690, 58755), (121470, 59005, 121690, 59225), (121470, 59475, 121690, 59695), (121470, 59945, 121690, 60165), (121470, 60415, 121690, 60635), (121470, 60885, 121690, 61105), (121470, 61355, 121690, 61575), (121470, 61825, 121690, 62045), (121470, 62295, 121690, 62515), (121470, 62765, 121690, 62985), (121470, 63235, 121690, 63455), (121470, 63705, 121690, 63925), (121470, 64175, 121690, 64395), (121470, 64645, 121690, 64865), (121470, 65115, 121690, 65335), (121470, 65585, 121690, 65805), (121470, 66055, 121690, 66275), (121470, 66525, 121690, 66745), (121470, 66995, 121690, 67215), (121470, 67465, 121690, 67685), (121470, 67935, 121690, 68155), (121470, 68405, 121690, 68625), (121470, 68875, 121690, 69095), (121470, 69345, 121690, 69565), (121470, 69815, 121690, 70035), (121470, 70285, 121690, 70505), (121470, 70755, 121690, 70975), (121470, 71225, 121690, 71445), (121470, 71695, 121690, 71915), (121470, 72165, 121690, 72385), (121470, 72635, 121690, 72855), (121470, 73105, 121690, 73325), (121470, 73575, 121690, 73795), (121470, 74045, 121690, 74265), (121470, 74515, 121690, 74735), (103800, 75430, 104020, 75650), (104270, 75430, 104490, 75650), (104740, 75430, 104960, 75650), (105210, 75430, 105430, 75650), (105680, 75430, 105900, 75650), (106150, 75430, 106370, 75650), (106620, 75430, 106840, 75650), (107090, 75430, 107310, 75650), (107560, 75430, 107780, 75650), (108030, 75430, 108250, 75650), (108500, 75430, 108720, 75650), (108970, 75430, 109190, 75650), (109440, 75430, 109660, 75650), (109910, 75430, 110130, 75650), (110380, 75430, 110600, 75650), (110850, 75430, 111070, 75650), (111320, 75430, 111540, 75650), (111790, 75430, 112010, 75650), (112260, 75430, 112480, 75650), (112730, 75430, 112950, 75650), (113200, 75430, 113420, 75650), (113670, 75430, 113890, 75650), (114140, 75430, 114360, 75650), (114610, 75430, 114830, 75650), (115080, 75430, 115300, 75650), (115550, 75430, 115770, 75650), (116020, 75430, 116240, 75650), (116490, 75430, 116710, 75650), (116960, 75430, 117180, 75650), (117430, 75430, 117650, 75650), (117900, 75430, 118120, 75650), (118370, 75430, 118590, 75650), (118840, 75430, 119060, 75650), (119310, 75430, 119530, 75650), (119780, 75430, 120000, 75650), (120250, 75430, 120470, 75650), (120720, 75430, 120940, 75650), (103050, 53755, 103270, 53975), (103050, 54225, 103270, 54445), (103050, 54695, 103270, 54915), (103050, 55165, 103270, 55385), (103050, 55635, 103270, 55855), (103050, 56105, 103270, 56325), (103050, 56575, 103270, 56795), (103050, 57045, 103270, 57265), (103050, 57515, 103270, 57735)]
shift_lst = (0, 2500, 0, 0)
fig, ax = plt.subplots()
base_center = get_bb_lst_center(bb_lst)
for bb in bb_lst:
min_x, min_y, max_x, max_y = bb
rect = mpathes.Rectangle((min_x, min_y), max_x - min_x, max_y - min_y, fill="", color='r')
ax.add_patch(rect)
new_bb_lst = stretch_data_lst(bb_lst, base_center, shift_lst)
for bb in new_bb_lst:
min_x, min_y, max_x, max_y = bb
rect = mpathes.Rectangle((min_x, min_y), max_x - min_x, max_y - min_y, fill="", color='blue')
ax.add_patch(rect)
plt.text(*base_center, s="center")
plt.axis('equal')
plt.show()
效果：
标签：75650,121470,bb,stretch,shift,pyhton,75430,121690,rectangle From： https://www.cnblogs.com/zxingwork/p/16652323.html