Skip to content

Commit f7ffe68

Browse files
committed
add parameter filter
1 parent 4b3a3e8 commit f7ffe68

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

dataretrieval/nwis.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -460,6 +460,13 @@ def get_gwlevels(
460460

461461
if datetime_index is True:
462462
df = format_datetime(df, "lev_dt", "lev_tm", "lev_tz_cd")
463+
464+
# Filter by kwarg parameterCd because the service doesn't do it
465+
if "parameterCd" in kwargs:
466+
pcodes = kwargs["parameterCd"]
467+
if isinstance(pcodes, str):
468+
pcodes = [pcodes]
469+
df = df[df["parameter_cd"].isin(pcodes)]
463470

464471
return format_response(df, **kwargs), NWIS_Metadata(response, **kwargs)
465472

0 commit comments

Comments
 (0)