Changing mob spawn rules with datapack?

For a custom map, I wish to have fish spawn in a superflat map's water. This would require changing their Y spawn location, presumably with a datapack
If I try looking it up, all I can find is how to make custom mobs. Is there any way to do it on vanilla mobs? I'm down to learning datapack making for it if I can find a proper tutorial