如何在Python中规范化colorbar？

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()
im = ax.imshow(np.random.rand(10, 10), cmap='viridis')

cbar = fig.colorbar(im, ax=ax)
cbar.ax.set_position([0.85, 0.1, 0.05, 0.8])  # 设置colorbar的位置和大小

cbar.set_label('Label', rotation=270)  # 设置colorbar的标签
cbar.set_ticks([0, 0.5, 1])  # 设置colorbar的刻度
cbar.set_ticklabels(['Low', 'Medium', 'High'])  # 设置colorbar的刻度标签

plt.show()