import matplotlib.pyplot as plt
import numpy as np


def display_result(vectors, clusters):
    colors = [np.random.rand(3, 1) for i in range(len(clusters))]
    for cluster_index, (centroid, cluster) in enumerate(clusters.items()):
        current_cluster = [vectors[i] for i in cluster]
        xs = list(map(lambda x: x[0], current_cluster))
        ys = list(map(lambda x: x[1], current_cluster))
        plt.scatter(xs, ys, c=colors[cluster_index])
        plt.scatter(centroid[0], centroid[1], c=colors[cluster_index], marker='x')
    plt.show()

import matplotlib.pyplot as plt
import numpy as np


def display_result(vectors, clusters):
    colors = [np.random.rand(3, 1) for i in range(len(clusters))]
    centroids_colors = [[1-x for x in color] for color in colors]
    for cluster_index, (centroid, cluster) in enumerate(clusters.items()):
        current_cluster = [vectors[i] for i in cluster]
        xs = list(map(lambda x: x[0], current_cluster))
        ys = list(map(lambda x: x[1], current_cluster))
        plt.scatter(xs, ys, c=colors[cluster_index])
        plt.scatter(centroid[0], centroid[1], c=centroids_colors[cluster_index], marker='x')
    plt.show()

Add drawing centroids with inverted colors
Add drawing centroids with inverted colors

  import matplotlib.pyplot as plt
  import numpy as np
  
  
  def display_result(vectors, clusters):
      colors = [np.random.rand(3, 1) for i in range(len(clusters))]
+     centroids_colors = [[1-x for x in color] for color in colors]
      for cluster_index, (centroid, cluster) in enumerate(clusters.items()):
          current_cluster = [vectors[i] for i in cluster]
          xs = list(map(lambda x: x[0], current_cluster))
          ys = list(map(lambda x: x[1], current_cluster))
          plt.scatter(xs, ys, c=colors[cluster_index])
-         plt.scatter(centroid[0], centroid[1], c=colors[cluster_index], marker='x')
+         plt.scatter(centroid[0], centroid[1], c=centroids_colors[cluster_index], marker='x')
      plt.show()
  
Add drawing centroids with inverted colors
## Code Before:
import matplotlib.pyplot as plt
import numpy as np


def display_result(vectors, clusters):
    colors = [np.random.rand(3, 1) for i in range(len(clusters))]
    for cluster_index, (centroid, cluster) in enumerate(clusters.items()):
        current_cluster = [vectors[i] for i in cluster]
        xs = list(map(lambda x: x[0], current_cluster))
        ys = list(map(lambda x: x[1], current_cluster))
        plt.scatter(xs, ys, c=colors[cluster_index])
        plt.scatter(centroid[0], centroid[1], c=colors[cluster_index], marker='x')
    plt.show()

## Instruction:
Add drawing centroids with inverted colors
## Code After:
import matplotlib.pyplot as plt
import numpy as np


def display_result(vectors, clusters):
    colors = [np.random.rand(3, 1) for i in range(len(clusters))]
    centroids_colors = [[1-x for x in color] for color in colors]
    for cluster_index, (centroid, cluster) in enumerate(clusters.items()):
        current_cluster = [vectors[i] for i in cluster]
        xs = list(map(lambda x: x[0], current_cluster))
        ys = list(map(lambda x: x[1], current_cluster))
        plt.scatter(xs, ys, c=colors[cluster_index])
        plt.scatter(centroid[0], centroid[1], c=centroids_colors[cluster_index], marker='x')
    plt.show()
