Scatterplot with different size, marker, and color from pandas dataframe

by

scatter can only do one kind of marker at a time, so you have to plot the different types separately. Fortunately pandas makes this easy:

import matplotlib.pyplot as plt
import pandas as pd
x = {'speed': [10, 15, 20, 18, 19],
     'meters' : [122, 150, 190, 230, 300],
     'type': ['phone', 'phone', 'gps', 'gps', 'car'],
     'weight': [0.2, 0.3, 0.1, 0.85, 0.0],
     'old': [1, 2, 4, 5, 8]}

m = pd.DataFrame(x)
mkr_dict = {'gps': 'x', 'phone': '+', 'car': 'o'}
for kind in mkr_dict:
    d = m[m.type==kind]
    plt.scatter(d.meters, d.speed, 
                s = 100* d.weight, 
                c = d.old, 
                marker = mkr_dict[kind])
plt.show()

enter image description here

…. Where’s the car? Well, the weight is 0.0 in the original test data, and we’re using weight for marker-size, so: can’t see it.

More Related Contents:

  1. plot different color for different categorical levels
  2. Color a scatter plot by Column Values
  3. How to add hovering annotations in matplotlib
  4. How to have clusters of stacked bars with python (Pandas)
  5. Setting different color for each series in scatter plot on matplotlib
  6. How to manually create a legend
  7. Matplotlib scatter plot legend
  8. Plotting grouped data in same plot using Pandas
  9. How to add hovering annotations to a plot
  10. How to draw vertical lines on a given plot
  11. Plot a histogram such that bar heights sum to 1 (probability)
  12. How to remove or hide x-axis labels from a seaborn / matplotlib plot
  13. Plotting time-series data with seaborn
  14. How to deal with NaN values when plotting a boxplot
  15. How can I use seaborn without changing the matplotlib defaults?
  16. Labeling boxplot in seaborn with median value
  17. Get data points from Seaborn distplot
  18. How to decrease the density of x-ticks in seaborn
  19. Using pandas crosstab to create a bar plot
  20. How to annotate a seaborn barplot with the aggregated value
  21. How to do a scatter plot with empty circles in Python?
  22. Discrete legend in seaborn heatmap plot
  23. seaborn heatmap using pandas dataframe
  24. heatmap-like plot, but for categorical variables in seaborn
  25. Sorted bar charts with pandas/matplotlib or seaborn
  26. Overlapping yticklabels: Is it possible to control cell size of heatmap in seaborn?
  27. How to color scatter markers as a function of a third variable
  28. How does one insert statistical annotations (stars or p-values)
  29. How do I align gridlines for two y-axis scales?
  30. Changing width of bars created with catplot or barplot

Leave a Comment