如何在Python中使用matplotlib创建子图

时间:2017-03-28 17:05:34

标签: python matplotlib

我写了这段代码,我的子图中有错误。我现在不知道我的代码有什么问题。你能救我吗?

import pywt
import scipy.io.wavfile as wavfile

import matplotlib.pyplot as plt

rate,signal = wavfile.read('a0025.wav')
time = [x /rate for x in range(0,len(signal))]
tree = pywt.wavedec(data=signal[:1000], wavelet='db2', level=4, mode='symmetric')
print(len(tree))
newTree = [tree[0]*0, tree[1]*0, tree[2]*0, tree[3]*0, tree[4]]
recSignal = pywt.waverec(newTree,'db2')
fig, ax = plt.subplot(2, 1)
ax[0].plot(time[:1000], signal[:1000])
ax[0].set_xlabel('Czas [s]')
ax[0].set_ylabel('Amplituda')
ax[1].plot(time[:1000], recSignal[:1000])
ax[1].set_xlabel('Czas [s]')
ax[1].set_ylabel('Amplituda')
plt.show()

错误:

 raise ValueError('Illegal argument(s) to subplot: %s' % (args,))
    ValueError: Illegal argument(s) to subplot: (2, 1)

2 个答案:

答案 0 :(得分:5)

由于错误明确指出,您将非法参数传递给pyplot.subplot()。如果您查看documentation for that function,您会看到它需要3个参数(可以压缩为一个参数):ax = plt.subplot(2, 1, 1)ax = plt.subplot(211)

但是,您要查找的功能是plt.subplots()(请注意最后的s),which generates both a figure and an array of subplots

f, (ax1, ax2) = plt.subplots(1, 2, sharey=True)
ax1.plot(x, y)
ax1.set_title('Sharing Y axis')
ax2.scatter(x, y)

答案 1 :(得分:1)

此错误似乎在文档中,请参见https://matplotlib.org/3.2.1/api/_as_gen/matplotlib.pyplot.subplots.html。他们在第三个示例中忘记了“ s”。前两个示例是正确的。
例如

# using tuple unpacking for multiple Axes
fig, (ax1, ax2) = plt.subplot(1, 2)
fig, ((ax1, ax2), (ax3, ax4)) = plt.subplot(2, 2)