save_numpy_array_to_image.py

import numpy as np
from PIL import Image
def save_array_as_image(array, filename):
    """Saves a 2D numpy array to an image"""
    image = Image.fromarray(array)
    image.save(filename)

    
array = np.random.random((512,512))*255
array = np.uint8(array)
print(array)
save_array_as_image(array, "image.png")