pandas – scatter plot with different color legend for each point

by

The following method will create a list of colors as long as your dataframe, and then plot a point with a label with each color:

import matplotlib.pyplot as plt
import matplotlib.cm as cm
import matplotlib.colors as colors
import numpy as np
import pandas as pd

fig, ax = plt.subplots()

df = pd.DataFrame({'n1':[1,2,1,3], 'n2':[1,3,2,1], 'l':['a','b','c','d']})

colormap = cm.viridis
colorlist = [colors.rgb2hex(colormap(i)) for i in np.linspace(0, 0.9, len(df['l']))]

for i,c in enumerate(colorlist):

    x = df['n1'][i]
    y = df['n2'][i]
    l = df['l'][i]

    ax.scatter(x, y, label=l, s=50, linewidth=0.1, c=c)

ax.legend()

plt.show()

enter image description here

More Related Contents:

  1. Clustered Stacked Bar in Python Pandas [closed]
  2. Pandas: Adding new column to dataframe which is a copy of the index column
  3. Color by Column Values in Matplotlib
  4. Plotting grouped data in same plot using Pandas
  5. How to add hovering annotations to a plot
  6. How to add multiple annotations to a barplot
  7. How to save the Pandas dataframe/series data as a figure?
  8. How to dynamically update a plot in a loop in IPython notebook (within one cell)
  9. matplotlib: plot multiple columns of pandas data frame on the bar chart
  10. multi index plotting
  11. pandas 0.21.0 Timestamp compatibility issue with matplotlib
  12. Legend only shows one label when plotting with pandas
  13. How to remove or hide x-axis labels from a seaborn / matplotlib plot
  14. Plotting time-series data with seaborn
  15. Line plot with data points in pandas
  16. Saving Matplotlib graphs to image as full screen
  17. Large Dataset Polynomial Fitting Using Numpy
  18. How to create a grouped bar plot
  19. Is it possible to add a string as a legend item in matplotlib
  20. Plot a bar plot from a Pandas DataFrame
  21. matplotlib plot window won’t appear
  22. Grouped Bar graph Pandas
  23. Using pandas crosstab to create a bar plot
  24. Change the facecolor of boxplot in pandas
  25. How to annotate a seaborn barplot with the aggregated value
  26. countplot() with frequencies
  27. Problem in combining bar plot and line plot
  28. Horizontal stacked bar chart in Matplotlib
  29. Plot pandas dates in matplotlib
  30. Plot all pandas dataframe columns separately

Leave a Comment