matplotlib 保存图片时覆盖的问题

版权声明：此文章版权归呆呆所有，转发请注明出处！ https://blog.csdn.net/zhangyu4863/article/details/82696421

def draw_img(r_list, len_):
    # 5图
    x_arr = []
    y_arr = []
    x_arr_white = []
    y_arr_white = []
    for j in range(1, len_ + 1):
        i = 0
        while r_list[len_ - j] >= i:
            r = int(r_list[len_ - j] * math.cos(60 * math.pi / 180))
            lst = [r - 4, r - 3, r - 2, r - 1, r, r + 1, r + 2, r + 3, r + 4]
            if i in lst:
                x_arr.append(i)
                y_arr.append(j)
            x_arr.append(-i)
            y_arr.append(j)
            i += 1
    x_arr = np.array(x_arr)
    y_arr = np.array(y_arr)
    x_arr_white = np.array(x_arr_white)
    y_arr_white = np.array(y_arr_white)
    plt.scatter(x_arr, y_arr, color='k')
    plt.scatter(x_arr_white, y_arr_white, color='y')
    plt.axis('off')
    plt.savefig(gl.gif_make_path + '5.png')

    # 4图
    x_arr = []
    y_arr = []
    x_arr_white = []
    y_arr_white = []
    for j in range(1, len_ + 1):  # 行循环
        i = 1
        while r_list[len_ - j] >= i:
            x_arr.append(i)
            y_arr.append(j)
            x_arr.append(-i)
            y_arr.append(j)
            i += 1
    x_arr = np.array(x_arr)
    y_arr = np.array(y_arr)
    x_arr_white = np.array(x_arr_white)
    y_arr_white = np.array(y_arr_white)
    plt.scatter(x_arr, y_arr, color='k')
    plt.scatter(x_arr_white, y_arr_white, color='y')
    plt.axis('off')
    plt.savefig(gl.gif_make_path + '4.png')

当你画了一张图后，保存后，当你要保存第二章图的时候会发现，第二张图会被第一张图覆盖，总之不是你想要的

这是因为你没有关闭之前的窗口

plt.close()  # 关闭窗口

所以上边的代码改为

def draw_img(r_list, len_):
    # 5图
    x_arr = []
    y_arr = []
    x_arr_white = []
    y_arr_white = []
    for j in range(1, len_ + 1):
        i = 0
        while r_list[len_ - j] >= i:
            r = int(r_list[len_ - j] * math.cos(60 * math.pi / 180))
            lst = [r - 4, r - 3, r - 2, r - 1, r, r + 1, r + 2, r + 3, r + 4]
            if i in lst:
                x_arr.append(i)
                y_arr.append(j)
            x_arr.append(-i)
            y_arr.append(j)
            i += 1
    x_arr = np.array(x_arr)
    y_arr = np.array(y_arr)
    x_arr_white = np.array(x_arr_white)
    y_arr_white = np.array(y_arr_white)
    plt.scatter(x_arr, y_arr, color='k')
    plt.scatter(x_arr_white, y_arr_white, color='y')
    plt.axis('off')
    plt.savefig(gl.gif_make_path + '5.png')
    plt.close()

    # 4图
    x_arr = []
    y_arr = []
    x_arr_white = []
    y_arr_white = []
    for j in range(1, len_ + 1):  # 行循环
        i = 1
        while r_list[len_ - j] >= i:
            x_arr.append(i)
            y_arr.append(j)
            x_arr.append(-i)
            y_arr.append(j)
            i += 1
    x_arr = np.array(x_arr)
    y_arr = np.array(y_arr)
    x_arr_white = np.array(x_arr_white)
    y_arr_white = np.array(y_arr_white)
    plt.scatter(x_arr, y_arr, color='k')
    plt.scatter(x_arr_white, y_arr_white, color='y')
    plt.axis('off')
    plt.savefig(gl.gif_make_path + '4.png')
    plt.close()