pulse2percept.viz.base

correlation_matrix, scatter_correlation

Functions

correlation_matrix(X[, cols, dropna, ax]) Plot feature correlation matrix (requires seaborn)
scatter_correlation(x, y[, marker, …]) Scatter plots some data points and fits a regression curve to them
pulse2percept.viz.base.correlation_matrix(X, cols=None, dropna=True, ax=None)[source]

Plot feature correlation matrix (requires seaborn)

New in version 0.7.

Parameters:
  • X (pd.DataFrame) – Data matrix as a Pandas DataFrame
  • cols (list or None) – List of columns to include in the correlation matrix
  • dropna ({True, False}) – Flag whether to drop columns or rows with NaN values
  • ax (matplotlib.axes.Axes or list thereof; optional, default: None) – A Matplotlib Axes object or a list thereof (one per electrode to plot). If None, a new Axes object will be created.
pulse2percept.viz.base.scatter_correlation(x, y, marker='o', marker_size=50, marker_alpha=0.5, color='k', text_size=10, show_slope_intercept=False, ax=None, autoscale=True)[source]

Scatter plots some data points and fits a regression curve to them

New in version 0.7.

Parameters:
  • y (x,) – x, y coordinates of data points to scatter
  • marker (str, optional) – Marker style passed to Matplotlib’s scatter
  • marker_size (float or array-like, shape (n, ), optional) – Marker size in points**2 passed to Matplotlib’s scatter
  • marker_alpha (float, optional) – Marker alpha value between 0 and 1
  • color (array-like or list of colors or color, optional) – Marker color passed to Matplotlib’s scatter
  • text_size (int, optional) – Font size for inset text and axis labels
  • ax (axis, optional) – Matplotlib axis
  • autoscale ({True, False}) – Flag whether to automatically adjust the axis limits