Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
if isinstance(threshold, int):
assert threshold > 0, 'threshold must be greater than zero'
df = df.iloc[threshold:]
if df.empty:
return df, None
cutoff_time = df.index[0]
elif isinstance(threshold, str):
if can_be_type(type=pd.tseries.frequencies.to_offset, string=threshold):
threshold = pd.tseries.frequencies.to_offset(threshold)
assert threshold.n > 0, 'threshold must be greater than zero'
cutoff_time = df.index[0] + threshold
elif can_be_type(type=pd.Timestamp, string=threshold):
cutoff_time = pd.Timestamp(threshold)
else:
raise ValueError('invalid threshold')
else:
is_timestamp = isinstance(threshold, pd.Timestamp)
assert is_timestamp, 'invalid threshold'
cutoff_time = threshold
if cutoff_time != df.index[0]:
df = df[df.index >= cutoff_time]
if df.empty:
return df, None
An offset will be applied relative to the first time in the index.
Returns:
df, cutoff_time (tuple(DataFrame, Timestamp)): Returns the data frame and the applied cutoff time.
"""
if isinstance(threshold, int):
assert threshold > 0, 'threshold must be greater than zero'
df = df.iloc[threshold:]
if df.empty:
return df, None
cutoff_time = df.index[0]
elif isinstance(threshold, str):
if can_be_type(type=pd.tseries.frequencies.to_offset, string=threshold):
threshold = pd.tseries.frequencies.to_offset(threshold)
assert threshold.n > 0, 'threshold must be greater than zero'
cutoff_time = df.index[0] + threshold
elif can_be_type(type=pd.Timestamp, string=threshold):
cutoff_time = pd.Timestamp(threshold)
else:
raise ValueError('invalid threshold')
else:
is_timestamp = isinstance(threshold, pd.Timestamp)
assert is_timestamp, 'invalid threshold'
cutoff_time = threshold
if cutoff_time != df.index[0]: