matplotlib 保存图片时覆盖的问题

深碍√TFBOYSˉ_ 2022-05-12 03:16 209阅读 0赞

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