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