| 25 | os.makedirs(FIG_DIR) |
| 26 | |
| 27 | def get_colors(n): |
| 28 | COLOR_FIVE = [ |
| 29 | "#f0f9e8", |
| 30 | "#bae4bc", |
| 31 | "#7bccc4", |
| 32 | "#43a2ca", |
| 33 | "#0868ac", |
| 34 | ] |
| 35 | COLOR_FIVE = [ |
| 36 | "#eff3ff", |
| 37 | "#bdd7e7", |
| 38 | "#6baed6", |
| 39 | "#3182bd", |
| 40 | "#08519c", |
| 41 | ] |
| 42 | COLOR_FOUR = ["#ca0020", "#f4a582", "#92c5de", "#0571b0"] |
| 43 | COLOR_THREE = [ |
| 44 | "#e0f3db", |
| 45 | "#a8ddb5", |
| 46 | "#43a2ca", |
| 47 | ] |
| 48 | COLOR_TWO = ["#ca0020", "#0571b0"] |
| 49 | # COLOR_TWO = ["#ef8a62", "#67a9cf"] |
| 50 | |
| 51 | COLOR_SIX = [ |
| 52 | "#b2182b", |
| 53 | "#ef8a62", |
| 54 | "#fddbc7", |
| 55 | "#d1e5f0", |
| 56 | "#67a9cf", |
| 57 | "#2166ac", |
| 58 | ] |
| 59 | COLOR_SEVEN = [ |
| 60 | "#eff3ff", |
| 61 | "#c6dbef", |
| 62 | "#9ecae1", |
| 63 | "#6baed6", |
| 64 | "#4292c6", |
| 65 | "#2171b5", |
| 66 | "#084594", |
| 67 | ] |
| 68 | |
| 69 | # COLOR_FOUR = ["#eff3ff", "#bdd7e7", "#6baed6", "#2171b5", ] |
| 70 | COLOR_THREE = [ |
| 71 | "#fee6ce", |
| 72 | "#fdae6b", |
| 73 | "#e6550d", |
| 74 | ] |
| 75 | |
| 76 | colors = { |
| 77 | 2: COLOR_TWO, |
| 78 | 3: COLOR_THREE, |
| 79 | 4: COLOR_FOUR, |
| 80 | 5: COLOR_FIVE, |
| 81 | 6: COLOR_SIX, |
| 82 | 7: COLOR_SEVEN, |
| 83 | } |
| 84 | return colors[n] |