Numpy 数组到图形

Question

我有一个简单的二维 Numpy 数组，由 0 和 1 组成。有没有一种简单的方法可以制作一个在相应坐标上有阴影的图形？

例如，如果我的数组是 [[1,0],[0,1]] 该图将是一个 2x2 的正方形，左上角和右下角的阴影为

Answer 1

您可以使用 matplotlib 为您绘制矩阵。

使用 matshow command with an appropriate colourmap 生成绘图。

例如

import numpy as np
import matplotlib.pyplot as plt

x = np.array([[1,0],[0,1]])
plt.matshow(x, cmap='Blues')
plt.show()

会产生：