Matplotlib：检查空图

Question

我有一个循环来加载和绘制一些数据，如下所示：

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

for filename in filenames:
    plt.figure()
    if os.path.exists(filename):
        x, y = np.loadtxt(filename, unpack=True)
        plt.plot(x, y)
    plt.savefig(filename + '.png')
    plt.close()

现在，如果文件不存在，则不会加载或绘制数据，但仍会保存（空）图形。在上面的例子中，我可以通过在if 语句中包含所有plt 调用来纠正这个问题。我的实际用例涉及更多，因此我正在寻找一种方法来询问matplotlib/plt/图形/轴是否图形/轴完全为空。类似的东西

for filename in filenames:
    plt.figure()
    if os.path.exists(filename):
        x, y = np.loadtxt(filename, unpack=True)
        plt.plot(x, y)
    if not plt.figure_empty():  # <-- new line
        plt.savefig(filename + '.png')
    plt.close()

Answer 1

检查斧头是否有使用plot()绘制的数据：

if ax.lines:

如果它们是使用scatter() 绘制的：

if ax.collections:

Answer 2

检查图中是否有带有fig.get_axes() 的轴是否适合您的目的？

fig = plt.figure()
if fig.get_axes():
    # Do stuff when the figure isn't empty.

Answer 3

正如您所说，显而易见的解决方案是在 if 语句中包含保存

for filename in filenames:
    plt.figure()
    if os.path.exists(filename):
        x, y = np.loadtxt(filename, unpack=True)
        plt.plot(x, y)
        plt.savefig(filename + '.png')  # <-- indentation here
    plt.close()

否则，这将取决于“空”的真正含义。如果是一个图形不包含任何坐标轴，

for filename in filenames:
    fig = plt.figure()
    if os.path.exists(filename):
        x, y = np.loadtxt(filename, unpack=True)
        plt.plot(x, y)
    if len(fig.axes) > 0:  
        plt.savefig(filename + '.png')
    plt.close()

但是，这些都是某种解决方法。我认为您真的想自己执行逻辑步骤。

for filename in filenames:
    plt.figure()
    save_this = False
    if os.path.exists(filename):
        x, y = np.loadtxt(filename, unpack=True)
        plt.plot(x, y)
        save_this = True
    if save_this:
        plt.savefig(filename + '.png')
    plt.close()