Kom igångKom igång gratis

Visualisera kluster med matplotlib

Vi har diskuterat att visualiseringar är nödvändiga för att utvärdera de kluster som bildas och identifiera mönster i dina data. Nu ska vi fokusera på att visualisera besökardata från Comic-Con med hjälp av modulen matplotlib.

Data lagras i en pandas DataFrame, comic_con. x_scaled och y_scaled är kolumnnamnen för de standardiserade X- och Y-koordinaterna för personer vid en given tidpunkt. cluster_labels innehåller klustetiketterna. Ett länkningsobjekt lagras i variabeln distance_matrix.

Den här övningen är en del av kursen

Klusteranalys i Python

Visa kurs

Övningsinstruktioner

  • Importera klassen pyplot från modulen matplotlib som plt.
  • Definiera en ordbok colors för två klustetikett-värden, 1 och 2.
  • Rita ett punktdiagram där varje kluster får en färg enligt definitionen i ordboken colors.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Import the pyplot class
____

# Define a colors dictionary for clusters
colors = {____:'red', ____:'blue'}

# Plot a scatter plot
comic_con.plot.scatter(x=____, 
                	   y=____,
                	   c=comic_con['cluster_labels'].apply(____))
plt.show()
Redigera och kör kod