minecraft-assets/data/minecraft/worldgen/placed_feature/patch_waterlily.json

{
  "feature": "minecraft:patch_waterlily",
  "placement": [
    {
      "type": "minecraft:count",
      "count": 4
    },
    {
      "type": "minecraft:in_square"
    },
    {
      "type": "minecraft:heightmap",
      "heightmap": "WORLD_SURFACE_WG"
    },
    {
      "type": "minecraft:biome"
    }
  ]
}