| @@ 2469-2480 (lines=12) @@ | ||
| 2466 | generator. |
|
| 2467 | """ |
|
| 2468 | ||
| 2469 | def voltage_levels(p: float) -> int: |
|
| 2470 | if p <= 100: |
|
| 2471 | return 7 |
|
| 2472 | elif p <= 200: |
|
| 2473 | return 6 |
|
| 2474 | elif p <= 5500: |
|
| 2475 | return 5 |
|
| 2476 | elif p <= 20000: |
|
| 2477 | return 4 |
|
| 2478 | elif p <= 120000: |
|
| 2479 | return 3 |
|
| 2480 | return 1 |
|
| 2481 | ||
| 2482 | # Join mastr table |
|
| 2483 | with db.session_scope() as session: |
|
| @@ 192-203 (lines=12) @@ | ||
| 189 | GeoDataFrame containing units all having assigned a voltage level. |
|
| 190 | """ |
|
| 191 | ||
| 192 | def voltage_levels(p: float) -> int: |
|
| 193 | if p <= 100: |
|
| 194 | return 7 |
|
| 195 | elif p <= 200: |
|
| 196 | return 6 |
|
| 197 | elif p <= 5500: |
|
| 198 | return 5 |
|
| 199 | elif p <= 20000: |
|
| 200 | return 4 |
|
| 201 | elif p <= 120000: |
|
| 202 | return 3 |
|
| 203 | return 1 |
|
| 204 | ||
| 205 | units_gdf["voltage_level_inferred"] = False |
|
| 206 | mask = units_gdf.voltage_level.isna() |
|