Automatic stationary conversion

def make_stationary(data: pd.Series, alpha: float = 0.05, max_diff_order: int = 10) -> dict:
    # Test to see if the time series is already stationary
    if adfuller(data)[1] < alpha:
        return {
            'differencing_order': 0,
            'time_series': np.array(data)
        }
    
    # A list to store P-Values
    p_values = []
    
    # Test for differencing orders from 1 to max_diff_order (included)
    for i in range(1, max_diff_order + 1):
        # Perform ADF test
        result = adfuller(data.diff(i).dropna())
        # Append P-value
        p_values.append((i, result[1]))
        
    # Keep only those where P-value is lower than significance level
    significant = [p for p in p_values if p[1] < alpha]
    # Sort by the differencing order
    significant = sorted(significant, key=lambda x: x[0])
    
    # Get the differencing order
    diff_order = significant[0][0]
    
    # Make the time series stationary
    stationary_series = data.diff(diff_order).dropna()
    
    return {
        'differencing_order': diff_order,
        'time_series': np.array(stationary_series)
    }

Source: gist.githubusercontent.com

Similar codes

0 vote175 views1 answer
Python program to read a random line from a file.
Python
-3 votes201 views1 answer
ValueError: With n_samples=0, test_size=0.2 and train_size=None, the resulting train set will be empty. Adjust any of the aforementioned parameters.
Python
5 votes312 views6 answers
Fibonacci series in python
Python
0 vote281 views1 answer
Solution to Remove Recursion Limitation in python
Python
0 vote201 views1 answer
How to let someone select a folder in python
Python
0 vote224 views1 answer
Get list of all document in django-elasticsearch-dsl
Python Django
1 vote235 views1 answer
No handles with labels found to put in legend
Python
0 vote200 views1 answer
Make value of two tables unique django
Python
1 vote223 views1 answer
GdScript int
Python
0 vote192 views1 answer
BeautifulSoup(raw_html
Python