input: Write a python function to perform KMeans clustering on a given dataset and return the cluster labels.
  assert kmeans_clustering([[2, 4], [5, 8], [3, 7], [9, 2], [7, 1]], 3)==[1, 2, 2, 0, 0]
  assert kmeans_clustering([[1, 2], [1, 4], [1, 0], [10, 2], [10, 4], [10, 0]], 2)==[1, 1, 1, 0, 0, 0]
  assert kmeans_clustering([[3, 4], [2, 8], [6, 7], [8, 2], [5, 1]], 3)==[2, 0, 1, 2, 2]
output: def kmeans_clustering(data, k):
      from sklearn.cluster import KMeans
      kmeans = KMeans(n_clusters=k)
      kmeans.fit(data)
      return kmeans.labels_.tolist()