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()

现在，如果文件不存在，则不会加载或绘制数据，但仍会保存（空）图形。在上面的示例中，我可以通过在plt语句中包含所有if调用来简单地解决此问题。我的实际用例涉及的程度更高，因此我正在寻找一种方法来询问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()

Matplotlib：检查空图

3 个答案: