python - 匀称多边形到二进制掩码
问题描述
我看过这个问题,但还没有真正找到完整的答案。我有一个简单的形状多边形,称为polygon
. 我想将此多边形提取为二进制掩码(最好是 numpy 数组)。我该怎么做呢?
我还设法从 shapely 转换为 geopandas,如此处所示,因此从 geopandas 中提取掩码也可以,但不幸的是,我还没有真正找到关于此的线程。
编辑:要清楚,如果我要使用坐标网格,我的网格包含 x 和 y 笛卡尔坐标(无序),对应于构成形状轮廓的点。这些是浮点数,因此需要 int 输入的解决方案将无法正常工作。理想情况下,我希望起点是一个匀称的多边形而不是一组点,但如果这更可取,我可以使用一组无序的点(或者以某种方式从匀称的多边形中提取顺时针顶点)
我已经尝试过这里描述的 Yusuke 的方法但是我得到的面具不太有意义。
佑介的方法:
#%% create grid and plot
nx, ny = 100, 100
poly_verts = Plane1verts #this is a list of tuples containing cartesian coordinate pairs of the shape contour in x and y
# Create vertex coordinates for each grid cell...
# (<0,0> is at the top left of the grid in this system)
x, y = np.meshgrid(np.arange(nx), np.arange(ny))
x, y = x.flatten(), y.flatten()
points = np.vstack((x,y)).T
path = Path(poly_verts)
grid = path.contains_points(points)
grid = grid.reshape((ny,nx))
plt.imshow(grid)
plt.title('Grid plot')
plt.show()
这不是我所期望的。而如下所述的 geopandas 绘图显示了正确的形状。
#%% create shapely and plot for comparison
from shapely.geometry import Polygon
#convert the sets of points dict to a shapely object
polygon1_plane1=Polygon(Plane1vert_tuple)
p = gpd.GeoSeries(polygon1_plane1)
p.plot()
plt.show()
EDIT2:这是我用作元组列表的坐标网格的副本
[(-8.982, -12.535), (-7.478, -12.535), (-5.975, -12.535), (-4.471, -12.535), (-4.471, -12.535), (-2.967, -11.031), (-1.463, -11.031), (-1.463, -11.031), (0.041, -9.527), (0.041, -9.527), (1.544, -8.023), (3.048, -8.023), (4.552, -8.023), (4.552, -8.023), (6.056, -6.52), (7.559, -6.52), (7.559, -6.52), (7.559, -5.016), (9.063, -3.512), (10.567, -3.512), (10.567, -3.512), (10.567, -2.008), (10.567, -0.505), (10.567, 0.999), (10.567, 2.503), (10.567, 4.007), (10.567, 4.007), (9.063, 5.51), (9.063, 5.51), (7.559, 7.014), (7.559, 7.014), (6.056, 8.518), (6.056, 8.518), (4.552, 10.022), (4.552, 11.526), (4.552, 11.526), (3.048, 11.526), (1.544, 11.526), (1.544, 11.526), (1.544, 10.022), (0.041, 8.518), (0.041, 8.518), (0.041, 7.014), (-1.463, 5.51), (-2.967, 5.51), (-4.471, 5.51), (-4.471, 5.51), (-5.975, 4.007), (-7.478, 4.007), (-8.982, 4.007), (-10.486, 4.007), (-11.99, 4.007), (-13.493, 4.007), (-13.493, 4.007), (-14.997, 2.503), (-14.997, 2.503), (-16.501, 0.999), (-18.005, 0.999), (-18.005, 0.999), (-18.005, -0.505), (-19.508, -2.008), (-19.508, -2.008), (-19.508, -3.512), (-19.508, -5.016), (-19.508, -5.016), (-18.005, -6.52), (-18.005, -8.023), (-18.005, -8.023), (-16.501, -9.527), (-16.501, -9.527), (-14.997, -9.527), (-13.493, -11.031), (-13.493, -11.031), (-11.99, -11.031), (-10.486, -12.535), (-10.486, -12.535)]
解决方案
rasterio.features.rasterize听起来正是您正在寻找的东西。
from shapely.geometry import Polygon
import rasterio.features
import matplotlib.pyplot as plt
poly = Polygon([(0, 50), (10, 10), (30, 0), (45, 45), (0, 50)])
img = rasterio.features.rasterize([poly], out_shape=(60, 50))
plt.imshow(img)
推荐阅读
- css - React Div 滚动绑定另一个 Div
- google-apps-script - Google 电子表格范围的屏幕截图
- three.js - 有没有办法修复类型'null'不能用作索引类型.ts(2538)
- c - 如何防止线程读取 DPDK RX 和 TX 缓冲区?
- swift - 捐赠快捷方式时的问题:无法捐赠与没有有效快捷方式类型的意图的交互
- transactions - 是否可以原子地向rabbitMQ发布两条消息?
- reactjs - html-react-parser:将 DOM 节点替换为另一个包含内部文本子节点的 DOM 节点(A-link)
- python - url_for() 不渲染子目录中的文件
- python - Keras 已安装但无法导入,因为它说找不到 keras 模块
- sql - 在没有子选择的情况下选择同一张表