Set color of a mesh based on a value in a 3D matrix

Godot Version

4.2.1

Question

I am new to shaders and I have a question about the following. I’ve got my data in a big x,y,z matrix and each value in that matrix corresponds to a specific color. So for example element matrix[1][1][1] has a value of 1 which stands for a green object and element matrix[1][1][0] has a value of 9 which stands for a blue object.

Is it possible to use shader code to set a color based on this matrix value?

Thanks in advance!