Skip to content

Commit 826e8c3

Browse files
committed
Fix index problems
1 parent 9babc39 commit 826e8c3

2 files changed

Lines changed: 480 additions & 10 deletions

File tree

edisgo/io/pypsa_csv_import.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,8 @@ def _assign_timeseries(
161161

162162

163163
def _parse_timestamps(series: pd.Series) -> pd.DatetimeIndex:
164-
return pd.to_datetime(series)
164+
idx = pd.DatetimeIndex(pd.to_datetime(series.values), name="snapshot")
165+
return idx.freq and idx or pd.DatetimeIndex(idx, freq=pd.infer_freq(idx), name="snapshot")
165166

166167

167168
# ---------------------------------------------------------------------------
@@ -240,6 +241,7 @@ def populate_edisgo_from_pypsa_csv(
240241

241242
# --- Buses --------------------------------------------------------- #
242243
buses_raw = _read(folder, "buses.csv")
244+
buses_raw = buses_raw[buses_raw["v_nom"] <= 20]
243245
buses = _keep_and_rename(buses_raw, BUSES_RENAME)
244246

245247
# Reproject coordinates from source CRS to WGS-84
@@ -374,6 +376,7 @@ def _build_hv_trafos(raw: pd.DataFrame, base: pd.DataFrame) -> pd.DataFrame:
374376

375377
# --- Loads --------------------------------------------------------- #
376378
loads_raw = _read(folder, "loads.csv")
379+
377380
loads = _keep_and_rename(loads_raw, LOADS_RENAME)
378381

379382
# p_set: static baseline, overwritten with time-series max where available.
@@ -416,7 +419,8 @@ def _build_hv_trafos(raw: pd.DataFrame, base: pd.DataFrame) -> pd.DataFrame:
416419
snap = _read(folder, "snapshots.csv", index_col=None)
417420
if not snap.empty:
418421
timeindex = _parse_timestamps(snap.loc[:, "snapshot"].iloc[ts_slice])
419-
ts.timeindex = timeindex
422+
ts.timeindex = pd.DatetimeIndex(timeindex, name="snapshot",
423+
freq=pd.infer_freq(timeindex))
420424
else:
421425
timeindex = None
422426

@@ -435,6 +439,7 @@ def _build_hv_trafos(raw: pd.DataFrame, base: pd.DataFrame) -> pd.DataFrame:
435439

436440
# Active power - loads
437441
load_p = _read_ts(folder, "loads-p_set.csv", ts_slice)
442+
#load_p = load_p * 10 # to force use of 14a
438443
if not load_p.empty:
439444
ts.loads_active_power = _assign_timeseries(load_p, timeindex)
440445

0 commit comments

Comments
 (0)