如何在2D中绘制1D数据？

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 =＆＃34; Reds＆＃34;：