matplotlib 循环通过地块，然后取选定数量地块的总和

3pmvbmvn 于 2023-11-22 发布在其他

关注(0)|答案(1)|浏览(133)

我试图创建一个函数，绘制EOF并通过每个模式循环。这部分很好，我没有麻烦得到12个EOF的图。
然而，现在我想对前三个图、前六个图、前九个图进行求和，然后在每行的末尾生成一个图，以显示前三个图、前六个图等的累积和。
附件是我的图现在的样子。我想在第四列中添加求和。

的数据
我尝试创建一个新的循环，它基本上是一个“if语句”，表示如果列是最后一列，则对先前的模式进行求和，并为这些模式添加一个子图（在第4列中）。
不幸的是，在这样做的时候，我没有看到我的图有任何变化。我试过改变为对i求和（我会解释为遍历每种模式），甚至显式调用每种模式，这最终会产生错误。我很难弄清楚如何从for循环中对eofs求和。
代码如下：

def GPH_EOFs_plot(eof12):
    fig = plt.figure(figsize=(12,8))
    for i in range(0,12):
        clevs = np.linspace(-75, 75, 11)
        proj = ccrs.Orthographic(central_longitude=-90, central_latitude=90)
        ncols = 4
        nrows = 3
        ax = plt.subplot(ncols,nrows,i+1, projection = proj)
        ax.coastlines(color='grey', linewidth=1.0)
        ax.set_global()
        eofs = eof12.sel(mode=i)
        eofs_2d = eofs.squeeze()
        eofs_2d.plot.contourf(ax=ax, levels=clevs, cmap=plt.cm.RdBu_r, transform=ccrs.PlateCarree(), add_colorbar=False)
        if ncols == 3:
            eof_sum = sum(i)
            eof_sum.plot.contourf(ax=ax, levels=clevs, cmap=plt.cm.RdBu_r, transform=ccrs.PlateCarree(), add_colorbar=True)

    plt.savefig("EOFs_fromfunction.png",format='png')

字符串

matplotlib

来源：https://stackoverflow.com/questions/77462922/looping-through-plots-then-taking-sum-of-a-select-number-of-plots

1条答案

按热度按时间

xj3cbfub1#

代码有一些问题，我无法测试，但这是一个尝试。
首先，您的索引i用于eof12.sel(mode=i)中的mode和子图索引。当您仅绘制模式时，这很好。现在，您有一个新列，索引将具有不同的含义。
我将它们命名为imode和isubplt。
接下来，每次i % 3 == 2都有一个额外的子图。这是你的标记。每次发生这种情况，isubplt都必须增加，但imode不会增加。
最后，您需要通过创建一个新的ax来设置第四列。
这里缺少的是计算和的代码。从你所说的，你想将eof_sum初始化为零，并将每个eof_2d添加到它。你必须填写此代码。

def GPH_EOFs_plot(eof12):
    fig = plt.figure(figsize=(12,8))
    eof_sum = ### TODO: initialize eof_sum to all zeros ###
    isubplt = 1 # subplots start at 1
    for imode in range(0,12):
        clevs = np.linspace(-75, 75, 11)
        proj = ccrs.Orthographic(central_longitude=-90, central_latitude=90)
        ncols = 4
        nrows = 4 # now you will have four lines
        ax = plt.subplot(ncols, nrows, isubplt, projection = proj)
        isubplt += 1 # skip to the next subplot
        ax.coastlines(color='grey', linewidth=1.0)
        ax.set_global()
        eofs = eof12.sel(mode=imode)
        eofs_2d = eofs.squeeze()
        eof_sum = ### TODO: accumulate this plot ###
        eofs_2d.plot.contourf(ax=ax, levels=clevs, cmap=plt.cm.RdBu_r, transform=ccrs.PlateCarree(), add_colorbar=False)
        if imode % 3 == 2:
            # create the fourth subplot
            ax = plt.subplot(ncols, nrows, isubplt, projection = proj)
            isubplt += 1 # skip to the next subplot
            ax.coastlines(color='grey', linewidth=1.0)
            ax.set_global()
            eof_sum.plot.contourf(ax=ax, levels=clevs, cmap=plt.cm.RdBu_r, transform=ccrs.PlateCarree(), add_colorbar=True)

    plt.savefig("EOFs_fromfunction.png",format='png')

字符串

赞(0）回复(0）举报 2023-11-22

我来回答

matplotlib 循环通过地块，然后取选定数量地块的总和

1条答案

相关问题

热门标签

最新问答