Как построить данные 1D в 2D?

Question

У меня есть массив, который состоит из n ^ 2 чисел. Мне нужно сделать из него n * n 2D-массив и затем отобразить данные.

Например, график этого массива 3 * 3:

[0, 1, 0, 
1, 2, 1, 
0, 2, 0]

должно выглядеть так:

plot_example_010121020

Как построить это, используя python? Если мне нужно ввести оси X и Y, как это сделать?

Заранее спасибо.

Answer 1

import numpy as np
import matplotlib.pyplot as plt

a = np.arange(1, 10)
s = int(np.sqrt(a.size))
b = a.reshape(s, s)

plt.matshow(b)
plt.show()

, который получает вас

>>> a
[1 2 3 4 5 6 7 8 9]

>>> b
[[1 2 3]
 [4 5 6]
 [7 8 9]]

и график выглядит так

и используя входной массив, который вы предоставили cmaps = "Reds":