Prettify your matrices and print them in human readable way. No dependencies
Just copy draw_array.py into your code base, it has no dependencies
And then pass your JAX or Numpy arrays in following fashion:
display_array([array_1, array_2],["purple","blue"])
or to display it with values:
display_with_values([array_1, array_2],["green","yellow"])
Examples
just diplay:
⎡ ⎤
⎢ . . . . . ⎥⎡ ⎤⎡ ⎤⎡ ⎤
⎢ . . 5x2 . . ⎥⎢ . . 5x1 . . ⎥⎢ . . 5x1 . . ⎥⎢2x1⎥
⎣ ⎦⎣ ⎦⎣ ⎦⎣ ⎦
⎡ ⎤
⎢ . . . . . ⎥⎡ ⎤⎡ ⎤⎡ ⎤
⎢ . . 5x2 . . ⎥⎢ . . 5x1 . . ⎥⎢ . . 5x1 . . ⎥⎢2x1⎥
⎣ ⎦⎣ ⎦⎣ ⎦⎣ ⎦
or with values (it may not work as intented with large matrices):
⎡ ⎤
⎢ 1.0 0.0 0.0 ⎥
⎡ ⎤⎢ 0.0 1.0 0.0 ⎥⎡ ⎤⎡ ⎤
⎢ 1.0 1.0 1.0 ⎥⎢ 0.0 0.0 1.0 ⎥⎢ 1.0 1.0 1.0 ⎥⎢ 0.5 0.5 0.5 ⎥
⎣ ⎦⎣ ⎦⎣ ⎦⎣ ⎦
⎡ ⎤
⎢ 1.0 0.0 0.0 ⎥
⎢ 0.0 1.0 0.0 ⎥
⎢ 0.0 0.0 1.0 ⎥
⎡ ⎤⎢ 1.0 1.0 0.0 ⎥⎡ ⎤⎡ ⎤
⎢ 1.5 1.5 1.5 ⎥⎢ 0.0 0.0 1.0 ⎥⎢ 1.5 1.5 1.5 3.0 1.5 ⎥⎢ 0.0 0.0 0.0 0.5 0.5 ⎥
⎣ ⎦⎣ ⎦⎣ ⎦⎣ ⎦
⎡ ⎤
⎡ ⎤⎢ 0.0 0.0 0.0 1.0 0.0 ⎥⎡ ⎤⎡ ⎤
⎢ 1.5 1.5 1.5 3.5 2.0 ⎥⎢ 1.0 0.0 0.0 1.0 1.0 ⎥⎢ 3.5 7.0 ⎥⎢ 0.5 0.5 ⎥
⎣ ⎦⎣ ⎦⎣ ⎦⎣ ⎦