矢量化 matplotlib 个轴

Question

我试图避免在某些 matplotlib 图中使用“for”循环。 为此，我尝试了以下代码：

t=np.linspace(0,1,1000)
signals=[]

for k in range(8):  
    signals.append(np.sin(2*np.pi*t*k))
signals=np.array(signals)
    
units=np.array(['V', 'V', 'V', 'V', 'A', 'A', 'A', 'A'])
fig, axes = plt.subplots(8, 1,figsize=(10, 16))
# axes.plot(t,signals[units=="V",:])

最后一条注释行会引发错误，因为轴是一个 numpy 数组，它没有属性“plot”。 我试图将每个斧头分配给每个信号的 plot。

我怎样才能使这条线矢量化？

谢谢。

Answer 1

您不能矢量化多个子图的创建，但可以使用循环：

fig, axes = plt.subplots(signals.shape[0], 1, figsize=(8, signals.shape[0] * 2.5))

for i in range(signals.shape[0]):
    axes[i].plot(signals[i, ...])
    axes[i].set_ylabel('x(t) (unit: V)')
    ...

您可以做的是 plot 将所有内容放入相同的轴中（只需使用ax.plot(signals.T) ）。