恢复matplotlib默认轴刻度

Question

我的问题是，如何在更改它们后恢复matplotlib默认轴刻度。例如，在下面的代码中，我绘制了1到9的数字平方，然后将yticks更改为[20,40,60]。在我改变它们之前，这个图的默认yticks是[0,10,20,30,40,50,60,70,80]。所以，从现在开始，我怎样才能带回这些默认的yticks？

import matplotlib.pyplot as plt 
import numpy as np 
x = np.arange(9) + 1 
y = x ** 2 
fig, ax1 = plt.subplots() 
ax1.plot(x, y) 
ax1.set_yticks([20, 40, 60]) 
plt.show() 

Answer 1

我已经找到了自己的问题的答案。正如matplotlib文档中所述，AutoLocator是大多数绘图的缺省刻度线定位器。要启用AutoLocator，请查看下面我的脚本的重新编辑版本。

import matplotlib.pyplot as plt 
import numpy as np 
from matplotlib.ticker import AutoLocator 
x = np.arange(9) + 1 
y = x ** 2 
fig, ax1 = plt.subplots() 
ax1.plot(x, y) 
ax1.set_yticks([20, 40, 60]) 
ax1.yaxis.set_major_locator(AutoLocator()) # solution 
plt.show()