from tools.attractions.apis import Attractions | |
import pandas as pd | |
import sys | |
import os | |
sys.path.append(os.path.abspath(os.path.join(os.getcwd(), ".."))) | |
os.chdir(os.path.dirname(os.path.abspath(__file__))) | |
unique_cities = Attractions(path="../../database/attractions/attractions.csv").data['City'].unique() | |
df = Attractions(path="../../database/attractions/attractions.csv").data | |
print(len(df)) | |
citySet = open('../../database/background/citySet.txt','r').read().split('\n') | |
cnt = 0 | |
for city in unique_cities: | |
if city not in citySet: | |
df = df[df['City'] != city] | |
print(len(df)) | |
df.to_csv('../../database/attractions/attractions2.csv', index=False) |