сохранить несколько пятиугольников как изображение на каждом шаге цикла for

Question

Мне нужно сохранять 4 пятиугольника в качестве изображения на каждом шаге, но оно сохраняет каждое изображение на предыдущем. Мне нужно, чтобы каждое изображение содержало один отдельный многоугольник.

import numpy as np
import matplotlib.pyplot as plt

pixels = 600
my_dpi = 100
num_geo=3

coord = np.array([[[-150, -200], [300, -200], [300, 0], [150, 200], [-150, 200]],
                  [[-300, -200], [200, -300], [200, -50], [200, 300], [-150, 200]],
                  [[-140, -230], [350, -260], [350, 0], [140, 200], [-180, 220]],
                  [[-180, -240], [370, -270], [370, 0], [170, 200], [-190, 230]]])


fig = plt.figure(num_geo,figsize=( pixels/my_dpi,  pixels/my_dpi),facecolor='k', dpi=my_dpi)  


plt.axes([0,0,1,1])

rectangle = plt.Rectangle((-300, -300), 600, 600, fc='k')
plt.gca().add_patch(rectangle)



for i in range(4):

    geo = coord[i, :, :]
    print(coord[i])

    polygon = plt.Polygon(coord[i],color='w')
    plt.gca().add_patch(polygon)
    plt.savefig('figure/%d.jpg' % i, dpi=my_dpi)
    plt.axis('off')
    plt.axis([-300,300,-300,300])
i =+ 1

Answer 1

Вам нужно создать фигуру и прямоугольник фона внутри для l oop, а затем закрыть график, используя plt.close() в конце l oop. Если вы не закроете график, эта же ось будет использоваться для построения нового многоугольника. Вот ваш код с небольшими корректировками:

import numpy as np
import matplotlib.pyplot as plt

pixels = 600
my_dpi = 100
num_geo=3

coord = np.array([[[-150, -200], [300, -200], [300, 0], [150, 200], [-150, 200]],
                  [[-300, -200], [200, -300], [200, -50], [200, 300], [-150, 200]],
                  [[-140, -230], [350, -260], [350, 0], [140, 200], [-180, 220]],
                  [[-180, -240], [370, -270], [370, 0], [170, 200], [-190, 230]]])

for i in range(4):
    geo = coord[i, :, :]
    print(coord[i])
    fig = plt.figure(num_geo,figsize=( pixels/my_dpi,  pixels/my_dpi),facecolor='k', dpi=my_dpi)  
    plt.axes([0,0,1,1])
    rectangle = plt.Rectangle((-300, -300), 600, 600, fc='k')
    plt.gca().add_patch(rectangle)
    polygon = plt.Polygon(coord[i],color='w')
    plt.gca().add_patch(polygon)
    plt.axis('off')
    plt.axis([-300,300,-300,300])
    plt.savefig('figure/%d.jpg' % i, dpi=my_dpi)
    plt.close()

---

Вот немного более питон c способ написать для l oop, который использует стиль объекта matplotlib. Таким образом, вам не нужно закрывать фигуру, поскольку новая фигура и новая ось создаются в начале l oop:

for index, c in enumerate(coord):
    fig, ax = plt.subplots(1, figsize=( pixels/my_dpi, pixels/my_dpi), dpi=my_dpi)  
    rectangle = plt.Rectangle((-300, -300), 600, 600, fc='k')
    ax.add_patch(rectangle)
    polygon = plt.Polygon(c, color='w')
    ax.add_patch(polygon)
    ax.axis('off')
    ax.axis([-300,300,-300,300])
    fig.savefig('figure/%d.jpg' % index, dpi=my_dpi)