import h5py def generate_map( base_elevation, obstacle_mask, traffic_flow, terrain_map, output_path, metadata={} ): with h5py.File(output_path, 'w') as f: f.create_dataset('elevation', data=base_elevation) f.create_dataset('occupancy', data=obstacle_mask) f.create_dataset('traffic_flow', data=traffic_flow) f.create_dataset('terrain_type', data=terrain_map) for k, v in metadata.items(): f.attrs[k] = v