973. K Closest Points to Origin - LeetCode Solution

class Solution:
    
    from math import sqrt
    
    def kClosest(self, points: List[List[int]], K: int) -> List[List[int]]:
        
        w = []
        
        for i in range(len(points)):
            w.append([sqrt(points[i][0] **2 + points[i][1]**2) , points[i]])
        
        w.sort()
        
        q= []
        
        for i in range(K):
            q.append(w[i][1])
        
        return q