Matplotlib 3.0 Cookbook

How to do it...

The following are the steps involved in plotting the image with a colorbar:

1. Load the dataset on winequality.csv, as follows:

wine_quality = pd.read_csv('winequality.csv', delimiter=';')corr_wine = wine_quality.corr()

2. Define the figure with the following command:

fig = plt.figure(1, (16., 6.))

3. Define the first subplot, plot the image and the colorbar, and set the x and y ticks using this code:

plt.subplot(131)plt.imshow(corr_wine, cmap='plasma')plt.colorbar()plt.xticks(range(len(corr_wine)),corr_wine.columns, rotation=75)plt.yticks(range(len(corr_wine)),corr_wine.columns)

4. Using the following code, define the second subplot, plot the image, define AxesDivider, plot the colorbar on the right, and set the x and y ticks: ...