使用 gridspec 添加图形

Question

我正在尝试创建一个类似于 this question.

为什么我只得到两个面板，即只有 gs2：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

def main():
   fig = plt.figure()
   gs1 = gridspec.GridSpec(1,4)
   gs2 = gridspec.GridSpec(2,4)

   for n in range(4):
      ax00 = plt.subplot(gs1[0,n])
      ax10 = plt.subplot(gs2[0,n])
      ax11 = plt.subplot(gs2[1,n])

      ax00.plot([0,0],[0,1*n],color='r')
      ax10.plot([0,1],[0,2*n],color='b')
      ax11.plot([0,1],[0,3*n],color='g')
   plt.show()

main()

这给了我这个：

最后我想要一个像这样的图：

我使用问题末尾的代码获得的。但是，我想要gs2.update(hspace=0) 给出的图的可移动性（我尝试使用gridspec 的原因）。 IE。我想删除最后一行和第二行之间的空格。

def whatIwant():
    f, axarr = plt.subplots(3,4)

    for i in range(4):
        axarr[0][i].plot([0,0],[0,1*i],color='r')
        axarr[1][i].plot([0,1],[0,2*i],color='b') #remove the space between those and be able to move the plots where I want
        axarr[2][i].plot([0,1],[0,3*i],color='g')
    plt.show()

Answer 1

这确实是其中一种情况，使用GridSpecFromSubplotSpec 是有意义的。也就是说，您将创建一个整体 GridSpec，其中包含一列和 2 行（以及 1 比 2 的高度比）。在第一行中，您将放置一个 GridSpecFromSubplotSpec，其中包含一行和 4 列。在第二行中，您将放置一个有两行和 4 列的行，另外指定一个 hspace=0.0 以便底部两行之间没有任何间距。

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec


fig = plt.figure()

gs = gridspec.GridSpec(2, 1, height_ratios=[1,2])
gs0 = gridspec.GridSpecFromSubplotSpec(1, 4, subplot_spec=gs[0], wspace=0.4)
gs1 = gridspec.GridSpecFromSubplotSpec(2, 4, subplot_spec=gs[1], hspace=0.0, wspace=0.4)

for n in range(4):
    ax00 = plt.subplot(gs0[0,n])
    ax10 = plt.subplot(gs1[0,n])
    ax11 = plt.subplot(gs1[1,n], sharex=ax10)
    plt.setp(ax10.get_xticklabels(), visible=False)
    ax00.plot([0,0],[0,1*n],color='r')
    ax10.plot([0,1],[0,2*n],color='b')
    ax11.plot([0,1],[0,3*n],color='g')
plt.show()

与链接问题答案中的解决方案相比，此解决方案的优势在于您不会重叠 GridSpec，因此无需考虑它们之间的关系。

---

如果您仍然对问题中的代码不起作用的原因感兴趣：
您需要使用两个不同的 GridSpec，每个都有总行数（在本例中为 3）；但随后仅填充第一个 GridSpec 的第一行和第二个 GridSpec 的后两行：

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

def main():
    fig = plt.figure()
    gs1 = gridspec.GridSpec(3,4)
    gs2 = gridspec.GridSpec(3,4, hspace=0.0)

    for n in range(4):
        ax00 = plt.subplot(gs1[0,n])
        ax10 = plt.subplot(gs2[1,n])
        ax11 = plt.subplot(gs2[2,n])

        ax00.plot([0,0],[0,1*n],color='r')
        ax10.plot([0,1],[0,2*n],color='b')
        ax11.plot([0,1],[0,3*n],color='g')
    plt.show()

main()