Skip to content

Instantly share code, notes, and snippets.

@dougfoo
Created September 12, 2020 15:15
# truncated version
def analyzeAndSave(tokyo, japan, call, local=False):
tokyoBase = tokyo[['Date','Gender','Age']]
tokyoBase = pd.get_dummies(tokyoBase, columns=["Gender","Age"])
tokyoBase = tokyoBase.assign(ct=1)
sumtokyo = tokyoBase.groupby('Date').agg('sum')
dailyTrend = sumtokyo['ct']
dailyTrend.index.name = "name"
dailyTrend.name = "Tokyo"
dailyTrend = dailyTrend.to_frame()
dailyTrend['7dayAvg'] = dailyTrend.rolling(7).mean()
if (local):
dailyTrend.reset_index().to_json('data/dailyTrend.json', orient="records")
else:
s3 = boto3.resource("s3")
s3.Bucket(BUCKET_NAME).put_object(Key='data/dailyTrend.json', Body=dailyTrend.reset_index().to_json(orient="records"), ACL='public-read-write')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment