| @@ 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: |
|
| @@ 130-141 (lines=12) @@ | ||
| 127 | GeoDataFrame containing units all having assigned a voltage level. |
|
| 128 | """ |
|
| 129 | ||
| 130 | def voltage_levels(p: float) -> int: |
|
| 131 | if p <= 100: |
|
| 132 | return 7 |
|
| 133 | elif p <= 200: |
|
| 134 | return 6 |
|
| 135 | elif p <= 5500: |
|
| 136 | return 5 |
|
| 137 | elif p <= 20000: |
|
| 138 | return 4 |
|
| 139 | elif p <= 120000: |
|
| 140 | return 3 |
|
| 141 | return 1 |
|
| 142 | ||
| 143 | units_gdf["voltage_level_inferred"] = False |
|
| 144 | mask = units_gdf.voltage_level.isna() |
|