import matplotlib.pyplot as plt
import numpy as np
# Define data
data = np.array([[0.1, 0.2, 0.3], [0.4, 0.5, 0.6], [0.7, 0.8, 0.9]])
# Create heatmap using Matplotlib
fig, ax = plt.subplots()
im = ax.imshow(data, cmap='coolwarm')
# Add color bar
cbar = ax.figure.colorbar(im, ax=ax)
# Set title and axis labels
ax.set_title("Example Heatmap")
ax.set_xlabel("X-axis Label")
ax.set_ylabel("Y-axis Label")
# Show plot
plt.show()