Diagtwig de dispersión de Python con colores correspondientes a cuerdas

Quiero hacer un diagtwig de dispersión con python matplotlib donde el color del punto debe corresponder con una cadena en particular de un archivo de datos, así que algo como esto:

data = np.genfromtxt('filename.txt', delimiter=',', dtype=None, names=['a', 'b', 'c']) plt.scatter(data['a'], data['b']) 

Si la primera columna del archivo ‘a’ es un flotante, la segunda columna ‘b’ es un flotante y la tercera columna ‘c’ es una cadena. La columna de cadena contiene 5 palabras diferentes que me gustaría trazar, ya que 5 colores diferentes es el diagtwig de dispersión. ¿Algunas ideas? ¡Muchas gracias!

Algo a lo largo de estas líneas debería hacer el truco:

 color_dict = { 'Allan':'red', 'Betty':'blue', 'Chris':'black', 'Diane':'green','Eugene':'purple' } plt.scatter( data['a'], data['b'], color=[ color_dict[i] for i in data['c'] ] )