Coloring entries in an Matrix/2D-numpy array?

user_11614536 0
Viewed 2473

I'm learning python3 and I'd like to print a matrix/2d-array which is color-coded (CLI). So let's say I'd like to assign each of these integers a certain background color, creating a mosaic-style look.

I've figured out how to fill a matrix of a given size with random integers, but I can't wrap my head around on how to continue from here on to achieve background coloring for each individual entry in the matrix, depending on its value. This is how far I've come:

from random import randint
import numpy as np

def generate():
    n = 10
    m = 0
    map = np.random.randint(4 + 1, size=(n, n))
    print(map)

    for element in np.nditer(map):
        # iterating over each column is probably not the way to go...


generate()

Is there a way to do this? I was thinking of iterating through every column of the matrix and check by several if conditions whether the entry is 0,1,2,3 or 4 and, based on the condition, append that value with a certain background color to a new matrix, but I assume there is a far more elegant way to do this...

python-3.x matrix numpy colors arrays
1 Answers

The following will print a colored output on console... 

>>> map = np.random.randint(4 + 1, size=(10, 10))
>>> def get_color_coded_str(i):
...    return "\033[3{}m{}\033[0m".format(i+1, i)
... 
>>> map_modified = np.vectorize(get_color_coded_str)(map)
>>> print("\n".join([" ".join(["{}"]*10)]*10).format(*[x for y in map_modified.tolist() for x in y]))
>>>

enter image description here

To add background color use following fn

>>> def get_color_coded_str(i):
...    return "\033[4{}m{}\033[0m".format(i+1, i)

enter image description here

from random import randint
import numpy as np

def get_color_coded_str(i):
    return "\033[3{}m{}\033[0m".format(i+1, i)

def get_color_coded_background(i):
    return "\033[4{}m {} \033[0m".format(i+1, i)

def print_a_ndarray(map, row_sep=" "):
    n, m = map.shape
    fmt_str = "\n".join([row_sep.join(["{}"]*m)]*n)
    print(fmt_str.format(*map.ravel()))

n = 10
m = 20
map = np.random.randint(4 + 1, size=(n, m))
map_modified = np.vectorize(get_color_coded_str)(map)
print_a_ndarray(map_modified)
back_map_modified = np.vectorize(get_color_coded_background)(map)
print("-------------------------------------------------------")
print_a_ndarray(back_map_modified, row_sep="")

PS: print function modified as suggested by @hpaulj

user_16175480
Related