Те же данные, сохраненные создавать различные образы - Python

Question

У меня есть в моем коде два способа сохранить данные изображения, один только сохранить его значения в оттенках серого, а другой для генерации тепла изображения карты:

def save_image(self, name): 
    """ 
    Save an image data in PNG format 
    :param name: the name of the file 
    """ 
    graphic = Image.new("RGB", (self.width, self.height)) 
    putpixel = graphic.putpixel 
    for x in range(self.width): 
     for y in range(self.height): 
      color = self.data[x][y] 
      color = int(Utils.translate_range(color, self.range_min, self.range_max, 0, 255)) 
      putpixel((x, y), (color, color, color)) 
    graphic.save(name + ".png", "PNG") 

def generate_heat_map_image(self, name): 
    """ 
    Generate a heat map of the image 
    :param name: the name of the file 
    """ 
    #self.normalize_image_data() 
    plt.figure() 
    fig = plt.imshow(self.data, extent=[-1, 1, -1, 1]) 
    plt.colorbar(fig) 
    plt.savefig(name+".png") 
    plt.close() 

Класс что представляет мои данные это:

class ImageData: 
def __init__(self, width, height): 
    self.width = width 
    self.height = height 
    self.data = [] 
    for i in range(width): 
     self.data.append([0] * height) 

Проходя те же данные для обоих методов

ContourMap.save_image ("ImagesOutput/Varia bilityOfGradients/ContourMap ") ContourMap.generate_heat_map_image (" ImagesOutput/VariabilityOfGradients/ContourMapHeatMap ")

Я получаю одно изображение повернуты по отношению к другому.

Метод 1:

Метод 2:

Я не понимаю, почему, но я должен это исправить.

Любая помощь будет оценена по достоинству. Спасибо заранее.

Answer 1

По-видимому, данные находятся в строчном формате, но вы выполняете итерацию, как если бы она была в формате столбца, который вращает все это на -90 градусов.

Быстрое исправление, чтобы заменить эту строку:

color = self.data[x][y] 

... с этим:

color = self.data[y][x] 

(Хотя предположительно data является массивом, так что вы действительно должны использовать self.data[y, x] вместо этого.)

Более четкое исправление:

for row in range(self.height): 
    for col in range(self.width): 
     color = self.data[row][col] 
     color = int(Utils.translate_range(color, self.range_min, self.range_max, 0, 255)) 
     putpixel((col, row), (color, color, color)) 

---

Это может быть не совсем ясно из документации pyplot, но если вы посмотрите на imshow, он объясняет, что он принимает массив объект, похожий по форме (п, м) и отображает его в качестве MxN изображения.