Wasserstein $K$-means for clustering probability distributions