def streamplot(
    uv_motion_field,
    ax=None,
    geodata=None,
    axis="on",
    streamplot_kwargs=None,
    map_kwargs=None,
    step=20,
):
    """
    Function to plot a motion field as streamlines.
    Wrapper for :func:`pysteps.visualization.motionfields.motion_plot` passing
    `plot_type="streamplot"`.

    .. _`streamplot_doc`: https://matplotlib.org/api/_as_gen/matplotlib.pyplot.streamplot.html

    Parameters
    ----------
    uv_motion_field: array-like
        Array of shape (2, m,n) containing the input motion field.
    streamplot_kwargs: dict, optional
      Optional dictionary containing keyword arguments for the quiver method.
      This argument is passed to
      See the `streamplot_doc`_ matplotlib's documentation.

    Other parameters
    ----------------
    See :py:func:`pysteps.visualization.motionfields.motion_plot`.

    Returns
    -------
    out: axis object
        Figure axes. Needed if one wants to add e.g. text inside the plot.
    """

    if streamplot_kwargs is None:
        streamplot_kwargs = dict()

    return motion_plot(
        uv_motion_field,
        plot_type="streamplot",
        ax=ax,
        geodata=geodata,
        axis=axis,
        step=step,
        plot_kwargs=streamplot_kwargs,
        map_kwargs=map_kwargs,
    )
Function to plot a motion field as streamlines.
Wrapper for :func:`pysteps.visualization.motionfields.motion_plot` passing
`plot_type="streamplot"`.

.. _`streamplot_doc`: https://matplotlib.org/api/_as_gen/matplotlib.pyplot.streamplot.html

Parameters
----------
uv_motion_field: array-like
Array of shape (2, m,n) containing the input motion field.
streamplot_kwargs: dict, optional
Optional dictionary containing keyword arguments for the quiver method.
This argument is passed to
See the `streamplot_doc`_ matplotlib's documentation.

Other parameters
----------------
See :py:func:`pysteps.visualization.motionfields.motion_plot`.

Returns
-------
out: axis object
Figure axes. Needed if one wants to add e.g. text inside the plot.
pysteps/visualization/motionfields.py