import pandas as pd
import altair as alt

cols = ["日期","重症","普通","轻型","无症状","重症累计","普通累计","轻型累计","无症状累计"]

data = [
    ['2021-07-21',0,5,6,8,0,5,6,8],
    ['2021-07-22',0,7,5,6,0,12,11,14],
    ['2021-07-23',0,9,3,4,0,21,14,18],
    ['2021-07-24',0,2,0,2,0,23,14,20],
    ['2021-07-25',0,11,27,1,2,32,41,13],
    ['2021-07-26',0,3,28,0,2,38,66,6],
    ['2021-07-27',0,24,23,1,4,68,81,2],
    ['2021-07-28',0,8,10,0,7,86,78,2],
    ['2021-07-29',0,3,10,1,8,94,82,1],
    ['2021-07-30',0,1,5,0,8,95,87,1],
    ['2021-07-31',0,2,12,0,7,114,83,0],
    ['2021-08-01',0,5,6,0,7,127,81,0],
    ['2021-08-02',0,2,3,0,6,132,82,0],
    ['2021-08-03',0,2,1,0,2,143,78,0],
    ['2021-08-04',0,1,3,0,1,149,77,0],
    ['2021-08-05',0,1,0,0,0,150,65,0],
    ['2021-08-06',0,1,0,0,0,148,63,0],
    ['2021-08-07',0,2,0,0,0,148,63,0],
    ['2021-08-08',0,0,0,0,0,148,59,0],
    ['2021-08-09',0,2,0,0,0,159,48,0],
    ['2021-08-10',0,0,0,0,0,157,47,0],
    ['2021-08-11',0,1,0,0,0,157,46,0],
    ['2021-08-12',0,0,0,0,0,154,45,0],
    ['2021-08-13',0,0,0,0,0,152,39,0],
    ['2021-08-14',0,0,0,0,1,147,36,0],
    ['2021-08-15',0,0,0,0,1,146,36,0],
    ['2021-08-16',0,0,0,0,1,142,33,0],
    ['2021-08-17',0,0,0,0,0,129,31,0],
]

alldf = pd.DataFrame(data=data,columns=cols)

df1 = alldf.melt(id_vars=["日期"],value_vars=["重症","普通","轻型","无症状"],var_name="症状",value_name="人数").sort_values(by=["日期"])

alt.Chart(df1).mark_line().encode(
    x='日期',
    y='人数',
    color='症状',
    strokeDash='症状',
    tooltip=['日期','人数','症状']
).properties(
    title='南京疫情 每日新增',
    width=700,
    height=300
).display(actions=False)
df2 = alldf.melt(id_vars=["日期"],value_vars=["重症累计","普通累计","轻型累计","无症状累计"],var_name="症状",value_name="人数").sort_values(by=["日期"])

alt.Chart(df2).mark_line().encode(
    x='日期',
    y='人数',
    color='症状',
    strokeDash='症状',
    tooltip=['日期','人数','症状']
).properties(
    title='南京疫情 每日累计',
    width=700,
    height=300
).display(actions=False)