figu中的人物

2024-10-04 03:16:31 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在试验用matplotlib画一幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅幅。因为正方形是最直接的画法,所以我从正方形开始。最后,我想为具有一定宽度的多边形编写一个生成器。在给定的示例中,这将是一个4角多边形,具有直角和宽度1。你知道吗

我当前的代码绘制了以下内容,这与预期的一样,也几乎与预期的一样。你知道吗

square in square in square

注意在2,22,3之间有一行代码,如果使用正确的算法而不是当前的代码,我认为可以删除这行代码。你知道吗

上面的总结是一个方形框,两个框中的振幅随1增加,假设较大的框在其余框的后面。你知道吗

我编写生成上述代码的方法,其实并不是一个函数。这是一个非常丑陋的点集合,恰好类似于空心正方形。你知道吗

import matplotlib.path as mpath
import matplotlib.patches as mpatches
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

INNER_AMPLITUDE = 1.0
OUTER_AMPLITUDE = 3.0

Path_in = mpath.Path
path_in_data = [
    (Path_in.MOVETO, (INNER_AMPLITUDE, -INNER_AMPLITUDE)),
    (Path_in.LINETO, (-INNER_AMPLITUDE, -INNER_AMPLITUDE)),
    (Path_in.LINETO, (-INNER_AMPLITUDE, INNER_AMPLITUDE)),
    (Path_in.LINETO, (INNER_AMPLITUDE, INNER_AMPLITUDE)),
    (Path_in.CLOSEPOLY, (INNER_AMPLITUDE, -INNER_AMPLITUDE)),
    ]
codes, verts = zip(*path_in_data)
path_in = mpath.Path(verts, codes)
patch_in = mpatches.PathPatch(path_in, facecolor='g', alpha=0.3)
ax.add_patch(patch_in)

x, y = zip(*path_in.vertices)
line, = ax.plot(x, y, 'go-')

Path_out = mpath.Path
path_out_data = [
    (Path_out.MOVETO, (OUTER_AMPLITUDE, -OUTER_AMPLITUDE)),
    (Path_out.LINETO, (-OUTER_AMPLITUDE, -OUTER_AMPLITUDE)),
    (Path_out.LINETO, (-OUTER_AMPLITUDE, OUTER_AMPLITUDE)),
    (Path_out.LINETO, (OUTER_AMPLITUDE, OUTER_AMPLITUDE)),
    (Path_out.LINETO, (OUTER_AMPLITUDE, OUTER_AMPLITUDE-INNER_AMPLITUDE)),
    (Path_out.LINETO, (-(OUTER_AMPLITUDE-INNER_AMPLITUDE), OUTER_AMPLITUDE-INNER_AMPLITUDE)),
    (Path_out.LINETO, (-(OUTER_AMPLITUDE-INNER_AMPLITUDE), -(OUTER_AMPLITUDE-INNER_AMPLITUDE))),
    (Path_out.LINETO, (OUTER_AMPLITUDE-INNER_AMPLITUDE, -(OUTER_AMPLITUDE-INNER_AMPLITUDE))),
    (Path_out.LINETO, (OUTER_AMPLITUDE-INNER_AMPLITUDE, OUTER_AMPLITUDE-INNER_AMPLITUDE)),
    (Path_out.LINETO, (OUTER_AMPLITUDE, OUTER_AMPLITUDE-INNER_AMPLITUDE)),
    (Path_out.CLOSEPOLY, (OUTER_AMPLITUDE, OUTER_AMPLITUDE-INNER_AMPLITUDE)),
    ]
codes, verts = zip(*path_out_data)
path_out = mpath.Path(verts, codes)
patch_out = mpatches.PathPatch(path_out, facecolor='r', alpha=0.3)
ax.add_patch(patch_out)
plt.title('Square in a square in a square')

ax.grid()
ax.axis('equal')
plt.show()

请注意,我认为这是代码审查的主题之外,因为我正在寻找扩展我的功能,而不仅仅是重新编写,这是最佳实践。我觉得我完全错了。第一件事。你知道吗

我应该如何使用matplotlib绘制具有一定宽度的多边形,假设多边形的外部将被具有相同形式和至少相同宽度的条带包围,并在内部完全填充?你知道吗


Tags: path代码in宽度matplotlibaxoutpatch
1条回答
网友
1楼 · 发布于 2024-10-04 03:16:31

纯粹在matplotlib中处理多边形可能相当乏味。幸运的是,有一个非常好的库用于此类操作:shapely。 就您的目的而言,parallel_offset函数是一种可行的方法。 您感兴趣的多边形的边界由ring1ring2ring3定义:

import numpy as np
import matplotlib.pyplot as plt
import shapely.geometry as sg
from descartes.patch import PolygonPatch

# if I understood correctly you mainly need the difference d here
INNER_AMPLITUDE = 0.1
OUTER_AMPLITUDE = 0.2
d = OUTER_AMPLITUDE - INNER_AMPLITUDE

# fix seed, for reproducability
np.random.seed(11111)

# a function to produce a "random" polygon
def random_polygon():
    nr_p = np.random.randint(7,15)
    angle = np.sort(np.random.rand(nr_p)*2*np.pi)
    dist = 0.3*np.random.rand(nr_p) + 0.5
    return np.vstack((np.cos(angle)*dist, np.sin(angle)*dist)).T

# your input polygon
p = random_polygon()

# create a shapely ring object
ring1 = sg.LinearRing(p)
ring2 = ring1.parallel_offset(d, 'right', join_style=2, mitre_limit=10.)
ring3 = ring1.parallel_offset(2*d, 'right', join_style=2, mitre_limit=10.)

# revert the third ring. This is necessary to use it to procude a hole
ring3.coords = list(ring3.coords)[::-1]

# inner and outer polygon
inner_poly = sg.Polygon(ring1)
outer_poly = sg.Polygon(ring2, [ring3])

# create the figure
fig, ax = plt.subplots(1)

# convert them to matplotlib patches and add them to the axes
ax.add_patch(PolygonPatch(inner_poly, facecolor=(0,1,0,0.4),
    edgecolor=(0,1,0,1), linewidth=3))
ax.add_patch(PolygonPatch(outer_poly, facecolor=(1,0,0,0.4),
    edgecolor=(1,0,0,1), linewidth=3))

# cosmetics
ax.set_aspect(1)
plt.axis([-1.5, 1.5, -1.5, 1.5])
plt.grid()
plt.show()

结果:

enter image description here

相关问题 更多 >