| Total Complexity | 6 |
| Total Lines | 26 |
| Duplicated Lines | 0 % |
| Changes | 0 | ||
| 1 | """ |
||
|
|
|||
| 2 | |||
| 3 | """ |
||
| 4 | |||
| 5 | |||
| 6 | class PlotUtils: |
||
| 7 | def _markers(self, df: pd.DataFrame, col: Optional[str]) -> Optional[pd.Series]: |
||
| 8 | from matplotlib.markers import MarkerStyle |
||
| 9 | |||
| 10 | if col is None: |
||
| 11 | return None |
||
| 12 | known = MarkerStyle.markers.keys() |
||
| 13 | uniques = df[col].unique() |
||
| 14 | if all((v in known for v in uniques)): |
||
| 15 | return df[col] |
||
| 16 | if len(uniques) > len(known): |
||
| 17 | logger.error(f"{len(uniques)} unique markers > {len(known)} available. Cycling.") |
||
| 18 | cycler = cycle(known) |
||
| 19 | dct = {k: next(cycler) for k in uniques} |
||
| 20 | return df[col].map(dct.get) |
||
| 21 | |||
| 22 | def _colors(df: pd.DataFrame, col: Optional[str]) -> Optional[pd.Series]: |
||
| 23 | if col is None: |
||
| 24 | return None |
||
| 25 | return df[col] |
||
| 26 |