array-show.py

import numpy as np
import pygame


def show(a):
    
    global screen
    
    if 'screen' not in globals():
        screen = pygame.display.set_mode((0, 0), pygame.RESIZABLE)

    for event in pygame.event.get():
        if event.type == pygame.VIDEORESIZE:
            screen = pygame.display.set_mode(event.dict['size'], pygame.RESIZABLE)

    pygame.transform.scale(pygame.surfarray.make_surface(
        np.swapaxes(a, 0, 1),
    ), screen.get_size(), screen)

    pygame.display.flip()