Как преобразовать массив 2d 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")