最接近原点的K个点(python)

题目描述：

我们有一个由平面上的点组成的列表 points。需要从中找出 K 个距离原点 (0,0) 最近的点。(这里，平面上两点之间的距离是欧几里德距离。)

你可以按任何顺序返回答案。除了点坐标的顺序之外，答案确保是唯一的。

示例1：

        输入：points = [[1,3],[-2,2]],K = 1

        输出：[[-2,2]]

        解释：(1, 3) 和原点之间的距离为 sqrt(10)， (-2, 2) 和原点之间的距离为 sqrt(8)， 由于 sqrt(8) < sqrt(10)，(-2, 2) 离原点更近。 我们只需要距离原点最近的 K = 1 个点，所以答案就是 [[-2,2]]。

示例2：

        输入：points = [[3,3],[5,-1],[-2,4]],K = 2

        输出：[[3,3],[-2,4]]

        (答案 [[-2,4],[3,3]] 也会被接受。)

class Solution(object):
    def kClosest(self, points, K):
        """
        :type points: List[List[int]]
        :type K: int
        :rtype: List[List[int]]
        """
        #思路：将points中每一组坐标点的平方和保存到第三个位置，根据第三列值进行排序，取出前K个进行处理输出
        for i in range(len(points)):
            points[i].append(points[i][0]**2 + points[i][1]**2)
            
        points = list(sorted(points,key=lambda x:x[2]))
        res = []
        
        for i in range(K):
            res.append([points[i][0],points[i][1]])
            
        return res

菜鸟一枚，代码仅供参考，如有问题，望指正~