matplotlib how to change data points color based on some variable

Question

I have 2 variables  x y  that change with time  t   I want to plot x vs  t and color the ticks based on the value of y  e g  for highest values of y the tick color is dark green  for lowest value is dark red  and for intermediate values the color will be scaled in between green and red   Can this be done with matplotlib in python

User · Accepted Answer

This is what matplotlib pyplot scatter is for   As a quick example   import matplotlib pyplot as plt import numpy as np    Generate data    t   np linspace 0  2   np pi  20  x   np sin t  y   np cos t   plt scatter t x c y  plt show

User · Answer

If you want to plot lines instead of points  see this example  modified here to plot good bad points representing a function as a black red as appropriate   def plot xx  yy  good          Plot data      Good parts are plotted as black  bad parts as red       Parameters     ----------     xx  yy   1D arrays         Data to plot      good    numpy ndarray   boolean         Boolean array indicating if point is good              import numpy as np     import matplotlib pyplot as plt     fig  ax   plt subplots       from matplotlib colors import from levels and colors     from matplotlib collections import LineCollection     cmap  norm   from levels and colors  0 0  0 5  1 5     red    black        points   np array  xx  yy   T reshape -1  1  2      segments   np concatenate  points  -1   points 1     axis 1      lines   LineCollection segments  cmap cmap  norm norm      lines set array good astype int       ax add collection lines      plt show

[python] matplotlib: how to change data points color based on some variable

Examples related to python

Examples related to matplotlib