-
Notifications
You must be signed in to change notification settings - Fork 0
/
reform_to_graph.py
72 lines (61 loc) · 1.93 KB
/
reform_to_graph.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# -*- coding: utf-8 -*-
import numpy as np
import plotly.graph_objs as go
def generate_graph__ppa(salaire_net, france_ppa, reform_ppa):
trace1 = go.Scatter(
name="Législation actuelle",
x=salaire_net,
y=france_ppa,
# fill='tonexty'
)
trace2 = go.Scatter(
name="Réforme",
x=salaire_net,
y=reform_ppa,
# fill='tozeroy'
)
data = [trace1, trace2]
layout = go.Layout(
title="Prime d'activité par salaire net",
# margin=go.Margin(l=50, r=50, b=50, t=50),
xaxis={'title': "salaire net (€ par mois)"},
yaxis={'title': "prime d'activité (€ par mois)"}
)
return go.Figure(data=data, layout=layout)
# TODO: clean france_ppa
def generate_graph__revenu_disponible(salaire_net, france_ppa, reform_ppa, reform_rsa, reform_revenu_disponible):
trace1 = go.Scatter(
name="RSA avec réforme",
x=salaire_net,
y=reform_rsa,
# mode='lines',
# line=dict(width=0.5,
# color='rgb(184, 247, 212)'),
# fill='tozeroy'
)
trace2 = go.Scatter(
name="Prime d'activité",
x=salaire_net,
y=np.array(reform_rsa) + np.array(reform_ppa),
# mode='lines',
# line=dict(width=0.5,
# color='rgb(111, 231, 219)'),
# fill='tozeroy'
)
trace3 = go.Scatter(
name="Revenu disponible",
x=salaire_net,
y=reform_revenu_disponible,
# mode='lines',
# line=dict(width=0.5,
# color='rgb(127, 166, 238)'),
# fill='tozeroy'
)
data = [trace1, trace2, trace3]
layout = go.Layout(
title="Prime d'activité, RSA & Revenu disponible",
# margin=go.Margin(l=50, r=50, b=50, t=50),
xaxis={'title': "salaire net (€ par mois)"},
yaxis={'title': "revenu disponible (€ par mois)"}
)
return go.Figure(data=data, layout=layout)