在numpy数组中绘制多边形

时间:2011-04-07 21:24:24

标签: python numpy

我正在尝试绘制这样的多边形:

In [1]: canvas = numpy.zeros((12, 12), dtype=int)

In [2]: mahotas.polygon.fill_polygon(
   ...: [(1, 1), (1, 10), (10, 10), (10, 1)],
   ...: canvas)

In [3]: canvas
Out[3]: 
array([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]])

我希望以下输出:

array([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]])

为什么[(10,2):(10:10)]仍然为零?是否有另一种方法可以将填充多边形绘制到数组中?

3 个答案:

答案 0 :(得分:1)

这是一个奇怪的结果。我发现如果你颠倒了点的顺序,它会绘制出完整的数字。换句话说:

# this is broken
pts = [(1, 1), (1, 10), (10, 10), (10, 1)]
# this works
pts = [(1, 1), (10, 1), (10, 10), (1, 10)]

这是一个测试程序:

import numpy
import mahotas.polygon

def run(n, reverse=0):
    canvas = numpy.zeros((n, n), dtype=int)
    lim = n-2
    print '\n%d x %d, lim=%d reverse=%d' % (n, n, lim, reverse)
    pts = [(1, 1), (1, lim), (lim, lim), (lim, 1), (1, 1)]
    if reverse:
        pts.reverse()
    mahotas.polygon.fill_polygon(pts, canvas)
    return canvas

for rev in (0, 1):
    for n in range(3, 14):
        print run(n, rev)

示例:

6 x 6, lim=4 reverse=0
[[0 0 0 0 0 0]
 [0 1 0 0 1 0]
 [0 1 1 1 1 0]
 [0 1 1 1 1 0]
 [0 1 0 0 0 0]
 [0 0 0 0 0 0]]

6 x 6, lim=4 reverse=1
[[0 0 0 0 0 0]
 [0 1 1 1 1 0]
 [0 1 1 1 1 0]
 [0 1 1 1 1 0]
 [0 1 1 1 1 0]
 [0 0 0 0 0 0]]

答案 1 :(得分:0)

如果多边形总是矩形,那么我们只需要两个点:

import numpy
canvas = numpy.zeros((12, 12), dtype=int)
points = [(1, 1), (1, 10), (10, 10), (10, 1)]
start_pt, end_pt = min(points), max(points)
canvas[start_pt[1]:end_pt[1]+1, start_pt[0]:end_pt[0]+1] = 1

答案 2 :(得分:0)

我认为反转是必需的,因为numpy将数组坐标引用为y,x(行,列),其中大多数其他程序将坐标作为x,y发出。