SAVGOL

Download Flojoy Studio to try this app

Apply a Savitzky-Golay filter to an input signal. This is generally used for smoothing data. The default behaviour is to implement a 3-point moving average of the data. Inputs ------ default : OrderedPair The data to apply the numpy savgol filter to. Params: window_length : int the length of the filter window, must be less than or equal to the size of the input poly_order : int the order of the polynomial used to fit the samples, must be less than or equal to the size of window_length Returns: out : OrderedPair x: time axis y: filtered signal

Python Code

import scipy
from flojoy import flojoy, OrderedPair
import warnings


@flojoy
def SAVGOL(
    default: OrderedPair, window_length: int = 50, poly_order: int = 1
) -> OrderedPair:
    """Apply a Savitzky-Golay filter to an input signal. This is generally used for smoothing data.

    The default behaviour is to implement a 3-point moving average of the data.

    Inputs
    ------
    default : OrderedPair
        The data to apply the numpy savgol filter to.

    Parameters
    ----------
    window_length : int
        the length of the filter window, must be less than or equal to the size of the input
    poly_order : int
        the order of the polynomial used to fit the samples, must be less than or equal to the size of window_length

    Returns
    -------
    OrderedPair
        x: time axis
        y: filtered signal
    """

    signal = default.y
    if window_length >= len(default.y):
        warnings.warn(
            "Polynomial order is greater than the window size. Using p=w-1..."
        )
        poly_order = len(default.y) - 1

    if poly_order >= window_length:
        warnings.warn(
            "Polynomial order is greater than the window size. Using p=w-1..."
        )
        poly_order = window_length - 1

    filtered = scipy.signal.savgol_filter(signal, window_length, poly_order)
    return OrderedPair(x=default.x, y=filtered)

Find this Flojoy Block on GitHub

Example App

Having problems with this example app? Join our Discord community and we will help you out!

Press enter or space to select a node.You can then use the arrow keys to move the node around. Press delete to remove it and escape to cancel.

Press enter or space to select an edge. You can then press delete to remove it or escape to cancel.

Download this Flojoy App

In this example, BASIC_OSCILLATOR generates a sine wave that has a sample rate of 30Hz over 2 seconds.

The output is then added to a RAND node that takes in a LINSPACE node to generate an array of 60 random numbers over 2 seconds. The result is a noisy signal resembling a sine wave.

To clean up the signal, the SAVGOL node is used with a window_length of 20 and a poly_order of 3, which smoothens out the noisy signal.