from PIL import Image


def is_prime(n):
    if n == 2:
        return True
    if n <= 1 or n % 2 == 0:
        return False
    for d in range(3, round(n ** 0.5) + 1, 2):
        if n % d == 0:
            return False
    return True


width, height = 1051, 446
n = 1
primes = 0
img = Image.new('L', (width, height), color=255)

for h in range(height):
    for w in range(width):
        if is_prime(n):
            img.putpixel((w, h), 0)
            primes += 1
        n += 1

img.save('Visualization of the first 39131 primes.png')
print(primes)  # prints 39131