在Python中循环将2D数组转换为RGB图像

Question

I want to convert a 2-dimensional array into an image in python using scipy.misc. Following is an example what I want to achieve

import numpy as np
from scipy.misc import toimage, imsave

L = 20      # one side of lattice
N = L*L     # number of sites in the lattice
xs = np.zeros((L,L), float)

x = []
for n in range(100):
    for i in range(L):
        for j in range(L):
            xs[i,j] = np.random.uniform(0.1, 0.9)
            x.append(xs)

In this code, for each n value I get a 2D array of xs which I am storing in list x. Instead of storing 2D array xs for each n in x, how can I convert it to an Colour RGB Image, where each value of x between 0.1 and 0.9 would be denoted by different colours ?

Answer 1

import numpy as np
import matplotlib.pyplot as plt

L = 20      # one side of lattice
N = L*L     # number of sites in the lattice
xs = np.zeros((L, L), float)

for n in range(100):
    for i in range(L):
        for j in range(L):
            xs[i,j] = np.random.uniform(0.1, 0.9)
            plt.imshow(xs)
plt.colorbar()
plt.savefig('temp.png')