Unfortunately, blue lightsabers and blue lightning become white.
Maybe those could be masked individually?
This could definitely be done, although it would take quite a bit of work to do it for every lightsaber shot throughout, especially if other things need to be masked out of the color adjustment as well.