python - 此函数在 Python 中的最短表示

Question

目前我有:

def func(points): #Input is a matrix with n lines and 2 columns.
    centroid = numpy.mean(points, axis=0)
    sum = 0
    for point in points:
        x = point[0] - centroid[0]
        y = point[1] - centorid[1]
        sum += x**2 + y**2
    return math.sqrt(sum)

Answer 1

有一个用于对序列求和的内置函数，称为 sum(很奇怪)。我们可以使用生成器理解创建数据，并将其直接提供给 sum 函数。因此:

return math.sqrt(sum((p[0]-centroid[0])**2 + (p[1]-centroid[1])**2 for p in points))