| import geopandas as gpd |
| import config |
| import numpy |
|
|
|
|
| def get(): |
| all_restaurants = gpd.read_file(config.restaurants_filepath) |
| all_restaurants = all_restaurants.dropna(subset=["name", "cuisine"]) |
| sub_restaurants = all_restaurants.sample( |
| config.total_restaurants_number, random_state=42 |
| ) |
| point_coordinates = [ |
| list([int(point.coords[0][0]) % 10000, int(point.coords[0][1]) % 10000]) |
| for point in sub_restaurants["geometry"].to_crs("epsg:2154") |
| ] |
| point_coordinates = numpy.array(point_coordinates) |
|
|
| return sub_restaurants, point_coordinates |
|
|
