import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.00， 3.50]
plt.rcParams["figure.autolayout"] = True

ax = plt.gca()
data = np.random.rand(6, 6)
ax.imshow(data)

#设置 xticks 和 yticks 
ax.set_xticks([1, 2, 3, 4, 5])
ax.set_yticks([1, 2, 3, 4, 5])

plt.show()