如何將二維 numpy 數組轉換為圖片？

Question

我還沒有在任何地方找到解決方案； 我找到了很多解決方案，可以通過不同顏色的圖片來可視化數組；

但是，我想要一些不同的東西：如果我有一個數組，讓我們讓它變得簡單，我有

a=np.ones([3,3])

然后我想將這個矩陣導出到 jpeg 或 png 格式，其中 numpy 數組看起來像來自紙張或書籍的漂亮矩陣，即它寫在 LaTeX 上； 所以在 LaTeX 中編譯命令時你會得到一些東西：

$a=\begin{pmatrix}1 & 1 & 1 \\\ 1 & 1 & 1 \\\ 1 & 1 & 1  \\\ \end{pmatrix}$

這可能嗎？

Answer 1

您可以使用 Matplotlib 將您的數組繪制為帶有plt.table的表格。 有關列標題的colLabels選項，請參閱鏈接的文檔。 此外，您可能想要修改圖形或軸來調整您的圖像。 對於以下示例，我僅使用plt.axis("off")關閉了軸。

import numpy as np
import matplotlib.pyplot as plt
data = np.random.randint(0, 1000, (5, 3))
plt.axis("off")
plt.table(cellText=data, colLabels=["A", "B", "C"], loc="center")
plt.savefig("data.png")