Pandas: easy visualization of correlation heatmap

df = pd.DataFrame

df.corr().style.background_gradient(cmap="coolwarm", axis=1)

# axis: 1 or ‘columns’ for column-wise, 0 or ‘index’ for row-wise