import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
import json
import folium
from folium import plugins
%matplotlib inline
import os
directory = './../Dataset5_MainTweetsFieldsAndScores/'
files = os.listdir(directory)
output_files = []
for _file in files:
if _file != '.ipynb_checkpoints':
output_files.append(os.path.join(directory, _file))
m = folium.Map()
count = 0
for idx in range(len(output_files)):
df = pd.read_csv(output_files[idx])
count += len(df)
def getColor(score):
if score > 0.01:
return 'green'
elif score <= 0.01 and score >= -0.01 :
return 'orange'
elif score < -0.01:
return 'red'
df['color'] = df['Sentiment Score'].apply(getColor)
for idx in range(len(df)):
folium.CircleMarker([df.iloc[idx, 3], df.iloc[idx, 2]],
color = df.iloc[idx, 4],
fill=True,
fill_opacity=0.7,
fill_color=df.iloc[idx, 4],
).add_to(m)
print(count)
m