Update kmeans.py

changed listcomp to map