df.style.background_gradient(cmap='coolwarm').set_precision(1)
I want to save the output of the code as an image