QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-05-09 01:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: linear imputation in Resample (#19393)

Browse Source 
(cherry picked from commit a39dd4493e)
This commit is contained in:
Yongjie Zhao
2022-03-28 22:30:45 +08:00
committed by Ville Brofeldt
parent 56e78b9ef7
commit 5f2ffb3ba4
4 changed files with 56 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
superset/utils/pandas_postprocessing/resample.py
View File

@@ -20,6 +20,7 @@ import pandas as pd
from flask_babel import gettext as _
from superset.exceptions import InvalidPostProcessingError
from superset.utils.pandas_postprocessing.utils import RESAMPLE_METHOD
def resample(
@@ -40,9 +41,15 @@ def resample(
"""
if not isinstance(df.index, pd.DatetimeIndex):
raise InvalidPostProcessingError(_("Resample operation requires DatetimeIndex"))
if method not in RESAMPLE_METHOD:
raise InvalidPostProcessingError(
_("Resample method should in ") + ", ".join(RESAMPLE_METHOD) + "."
)
if method == "asfreq" and fill_value is not None:
_df = df.resample(rule).asfreq(fill_value=fill_value)
elif method == "linear":
_df = df.resample(rule).interpolate()
else:
_df = getattr(df.resample(rule), method)()
return _df