How to change Root Locus plot color #789

Activity

1. roryyorke commented on Nov 13, 2022

   

   roryyorke

   on Nov 13, 2022

   Contributor

   More actions

   This was already asked for in #350.

   For now, you can findobj the lines after plotting, and then set the colours:

   import numpy as np
   import matplotlib.pyplot as plt
   
   import control as ct
   
   # bit of a hack
   def findlines(artist):
       return (isinstance(artist, plt.Line2D)
               and artist.get_linestyle()=='-'
               and isinstance(artist.get_xdata(), np.ndarray))
   
   g = ct.tf([1],[1,1]) * ct.tf([1,3],[1,2]) * ct.tf([1],[1,1,16])
   
   ct.rlocus(g);
   
   h = plt.gca().findobj(match=findlines)
   
   for i, h_ in enumerate(h): h_.set_color(f'C{i}')
   
   plt.show()

   
2. python-control

   locked and limited conversation to collaborators on Nov 27, 2022
3. murrayrm

   converted this issue into a discussion #803 on Nov 27, 2022