Representing array elements on a square in Python (as image)

Question

I have an array A with shape (3,3). Is there a way to represent the array elements on a square of size 3x3? In general, I would like to represent nxn arrays on a square of size nxn? The expected output is attached.

import numpy as np

A=np.array([[10,20,30],[40,50,60],[70,80,90]])

The expected output is:

Output

I want an image with array elements mentioned in the above format. — Wiz123
– Wiz123, Commented Jun 28, 2022 at 15:17
If you want an image you can go with matplotlib annotated heatmaps. — paime
– paime, Commented Jun 28, 2022 at 15:18

mozway · Accepted Answer · 2022-06-28 15:26:35Z

2

You could use seaborn.heatmap that has a nice API:

import numpy as np
import seaborn as sns
from matplotlib.colors import ListedColormap

A = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]])

ax = sns.heatmap(A,
            annot=True, square=True, cbar=False,
            xticklabels=False, yticklabels=False,
            cmap=ListedColormap(['white']),
            linecolor='k', lw=2,
            annot_kws={'size': 30}
           )

ax.figure.savefig('img.png')

output:

answered Jun 28, 2022 at 15:26

mozway

267k13 gold badges56 silver badges106 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Representing array elements on a square in Python (as image)

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related