Могу ли я сгенерировать изображение с помощью Python в Jupyter

Question

Есть ли способ использовать код Python для создания изображения (в памяти) и отображения его в блокноте Jupyter?

Я хочу отображать сетку пикселей, поэтому мне нужно иметь возможностьчтобы установить его размер и иметь возможность выбрать, какой из его пикселей светится, а какой нет.

в конце концов я бы хотел, чтобы он отображал анимированную серию изображений с вычислением пикселей.

Я пыталсяиспользуя:

import pygame as pg


pg.init()
screen= pg.display.set_mode([225,150])
r=pg.Color('red')
w=pg.Color('white')
data=[
    [r,r,r,r,r,r,r,r,r],
    [r,w,w,w,w,w,w,w,r],
    [r,w,w,w,w,w,w,w,r],
    [r,w,w,w,w,w,w,w,r],
    [r,w,w,w,w,w,w,w,r],
    [r,r,r,r,r,r,r,r,r]
]
for y,row in enumerate(data):
    for x,colour in enumerate(row):
        rect = pg.Rect(x*25,y*25,25,25)
        screen.fill(colour,rect=rect)

pg.display.update()
while True:
    for event in pg.event.get():
        if event.type == QUIT:
            pg.quit()
            sys.exit()
    CLOCK.tick(30)

Который я взял с https://www.youtube.com/watch?v=b84EywkQ3HI, но он, похоже, ничего не делает в jupyter.