I am trying to plot a heatmap using seaborn. So, I have values from 0 to 5 and I need to make a blue-to-red color scale, with white being at 1, blue- below 1, and red - from
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
data = np.random.rand(10,10)*5
ax = sns.heatmap(data, vmin=0,vmax=5,center=1,cmap="RdBu_r")
plt.show()