Add a random choice column to GeoJSON file

random_choice_to_geojson.py

import geopandas as gpd
import numpy as np
# Set a seed for reproducibility
np.random.seed(0)

input_path = 'input/path/to/geojson'
output_path = 'output/path/to/geojson'

choices = [1, 2, 3]

with open(output_path, 'w') as f:
    geojson = (gpd.read_file(input_path)
                  .assign(probafraud=lambda gdf: np.random.choice(choices, len(gdf.index)))
                  .to_json())
    f.write(geojson)