Я использую matplotlib 2.2.3, вот мой код ниже
import matplotlib.pyplot as plt
"""Plot out the different clusters"""
# Choose a different colour for each cluster
colour = plt.cm.rainbow(np.linspace(0, 1, len(centroids)))
Он выбрасывает не может найти rainbow ссылка