973. K Closest Points to Origin

https://leetcode.com/problems/k-closest-points-to-origin/

Python

import heapq
import math

class Solution:
    def kClosest(self, points: List[List[int]], k: int) -> List[List[int]]:
        return [
            coord for coord, distance
            in heapq.nsmallest(
                k,
                [((x,y), math.sqrt(x**2+y**2) ) for x, y in points],
                key=lambda item: item[1]
            )
        ]