Python 等同于 Mathematica 中的 ArrayPlot?

Python equivalent of ArrayPlot from Mathematica?

经过多次谷歌搜索,我决定只问这个问题。

我想要以下功能,但我完全不知道该怎么做。

https://reference.wolfram.com/language/ref/ArrayPlot.html

基本上,我想生成一个网格图,其中每个像素的颜色由我指定。

谢谢!

matplotlib包中,有一个函数imshow