如何在python中用形状计算重心?

时间:2018-11-27 14:33:16

标签: python shapely shapely.geometry

我发现了身材匀称,但是却找不到如何计算多边形的重心!

有人解决方案吗?

3 个答案:

答案 0 :(得分:2)

没有shapely的解决方案

您也可以手动计算,只需使用 numpy

import numpy as np

def polygon_area(xs, ys):
    """https://en.wikipedia.org/wiki/Centroid#Of_a_polygon"""
    # https://stackoverflow.com/a/30408825/7128154
    return 0.5 * (np.dot(xs, np.roll(ys, 1)) - np.dot(ys, np.roll(xs, 1)))

def polygon_centroid(xs, ys):
    """https://en.wikipedia.org/wiki/Centroid#Of_a_polygon"""
    xy = np.array([xs, ys])
    c = np.dot(xy + np.roll(xy, 1, axis=1),
               xs * np.roll(ys, 1) - np.roll(xs, 1) * ys
               ) / (6 * polygon_area(xs, ys))
    return c




print(polygon_centroid(xs=[0, 1, 1, 0], ys=[0, 0, 1, 1]),
      'expect: [.5, .5]')
print(polygon_centroid(xs=[0, 0, 2], ys=[1, -1, 0]),
      'expect: [2/3, 0]')
print(polygon_centroid(xs=[0, 0, 2], ys=[-1, 1, 0]),
      'expect: [2/3, 0]')

# https://wolfram.com/xid/0e5bspgmqyj9a5-cfx5ps
print(polygon_centroid(xs=[0, 1, 1.5, 1, 0, -.5], ys=[0, 0, .5, 1, 1, .5]),
      'expect: [.5, .5]')

输出

[0.5 0.5] expect: [.5, .5]
[ 0.66666667 -0.        ] expect: [2/3, 0]
[0.66666667 0.        ] expect: [2/3, 0]
[0.5 0.5] expect: [.5, .5]

答案 1 :(得分:1)

如果多边形的密度均匀,则其质心与centroid重合。在形状上,质心可以直接计算为:

from shapely.geometry import Polygon

P = Polygon([[0, 0], [1, 0], [1, 1], [0, 1]])

print(P.centroid)
#POINT (0.5 0.5)

答案 2 :(得分:0)

以上答案是正确的。但是有时候您不想使用这种格式。因此,您可以使用以下值:

from shapely.geometry import Polygon
centroid =   list(Polygon([[0, 0], [1, 0], [1, 1], [0, 1]]).centroid.coords)
#[(0.5, 0.5)]

我已经针对更复杂的几何体测试了该方法,并且效果很好。