Comment convertir un tableau numérique en (et afficher) une image?

Question

J'ai créé un tableau thusly:

import numpy as np 
data = np.zeros((512,512,3), dtype=np.uint8) 
data[256,256] = [255,0,0] 

Ce que je veux que ce faire est d'afficher un seul point rouge au centre d'une image 512x512. (Au moins pour commencer ... Je pense que je peux comprendre le reste à partir de là)

Answer 1

Vous pouvez utiliser PIL pour créer (et l'affichage) une image:

from PIL import Image 
import numpy as np 

w, h = 512, 512 
data = np.zeros((h, w, 3), dtype=np.uint8) 
data[256, 256] = [255, 0, 0] 
img = Image.fromarray(data, 'RGB') 
img.save('my.png') 
img.show() 

Answer 2

En utilisant pygame, vous pouvez ouvrir une fenêtre, obtenir la surface sous forme de tableau de pixels et la manipuler comme vous le souhaitez. Cependant, vous devrez copier votre tableau numpy dans le tableau de surface, ce qui sera beaucoup plus lent que de faire des opérations graphiques réelles sur les surfaces de pygame elles-mêmes.

Answer 3

Le Python Imaging Library peut afficher des images à l'aide de tableaux Numpy. Jetez un oeil à cette page pour exemple de code:

• Convert Between Numerical Arrays and PIL Image Objects

EDIT: Comme la note en bas de cette page dit, vous devriez vérifier les dernières notes de version qui rendent ce beaucoup plus simple:

http://effbot.org/zone/pil-changes-116.htm

Answer 4

Shortest Path est d'utiliser scipy, comme ceci:

from scipy.misc import toimage 
toimage(data).show() 

Cela nécessite également d'installer PIL ou Oreiller.

Une approche similaire aussi exigeant PIL ou oreiller, mais qui peut invoquer un autre spectateur est:

from scipy.misc import imshow 
imshow(data) 

Answer 5

-vous simplement dire cela?

from matplotlib import pyplot as plt 
plt.imshow(data, interpolation='nearest') 
plt.show()