vtkImageMapToColors should do some of what you want. Here’s the c++ example (no python example but more or less the same thing), https://vtk.org/Wiki/VTK/Examples/Cxx/Images/ImageMapToColors . Give it your vtkImageData and a color function, I prefer to use vtkColorTransferFunction for the lookuptable.