python中怎么用matplotlib在一行里生成多个图？

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 1, 100)
plt.figure(figsize=(10, 4))
# 一共有一行三列，现在画第一幅图
plt.subplot(1, 3, 1)
plt.plot(x, x ** 0.5)
# 一共有一行三列，现在画第二幅图
plt.subplot(1, 3, 2)
plt.plot(x, x ** 1)
# 一共有一行三列，现在画第三幅图
plt.subplot(1, 3, 3)
plt.plot(x, x ** 2)
plt.show()