code: def plot_lags(series, lags=1, suptitle=None):
      """Plot one or more lagged versions of a time series.
  
      Parameters
      ----------
      series : pd.Series
          Time series for plotting lags.
      lags : int or array-like, default=1
          The lag or lags to plot.
  
          - int plots the specified lag
          - array-like  plots specified lags in the array/list
  
      suptitle : str, default=None
          The text to use as the Figure's suptitle. If None, then the title
          will be "Plot of series against lags {lags}"
  
      Returns
      -------
      fig : matplotlib.figure.Figure
  
      axes : np.ndarray
          Array of the figure's Axe objects
  
      Examples
      --------
      >>> from sktime.utils.plotting import plot_lags
      >>> from sktime.datasets import load_airline
      >>> y = load_airline()
      >>> fig, ax = plot_lags(y, lags=2) # plot of y(t) with y(t-2)  # doctest: +SKIP
      >>> fig, ax = plot_lags(y, lags=[1,2,3]) # y(t) & y(t-1), y(t-2).. # doctest: +SKIP
      """
      _check_soft_dependencies("matplotlib")
      import matplotlib.pyplot as plt
  
      check_y(series)
  
      if isinstance(lags, int):
          single_lag = True
          lags = [lags]
      elif isinstance(lags, (tuple, list, np.ndarray)):
          single_lag = False
      else:
          raise ValueError(
              "`lags should be an integer, tuple, list, or np.ndarray."
          )
  
      length = len(lags)
      n_cols = min(3, length)
      n_rows = math.ceil(length / n_cols)
      fig, ax = plt.subplots(
          nrows=n_rows,
          ncols=n_cols,
          figsize=(8, 6 * n_rows),
          sharex=True,
          sharey=True,
      )
      if single_lag:
          axes = ax
          pd.plotting.lag_plot(series, lag=lags[0], ax=axes)
      else:
          axes = ax.ravel()
          for i, val in enumerate(lags):
              pd.plotting.lag_plot(series, lag=val, ax=axes[i])
  
      if suptitle is None:
          fig.suptitle(
              f"Plot of series against lags {', '.join([str(lag) for lag in lags])}",
              size="xx-large",
          )
      else:
          fig.suptitle(suptitle, size="xx-large")
  
      return fig, np.array(fig.get_axes())
  
file_path: sktime/sktime/utils/plotting.py
imports: import math
  from warnings import simplefilter, warn
  import numpy
  import pandas
  from sktime.datatypes import convert_to
  from sktime.utils.validation._dependencies import _check_soft_dependencies
  from sktime.utils.validation.forecasting import check_interval_df, check_y
  from sktime.utils.validation.series import check_consistent_index_type
label_desc: Use image node_id 3 for calling a global function with example usage: plot_lags(series, lags, suptitle) and returns: fig,  np