Draw an image from array with PIL

Question

I\'m trying to convert an array to an image using the Image.fromarray in PIL but i can\'t seem to get the shape I\'m looking for. Here\'s my code: