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 |