import matplotlib.pyplot as plt
import matplotlib as mp
import numpy as np
xs = "ABCDEFGHI"
ys = [5, 6, 7, 8, 9, 10, 11, 12, 13]
#Colorize the graph based on likeability:
likeability_scores = np.array([
5, 4.5, 3.5,
2.5, 1.5, .5,
2, 3, 4,
])
data_normalizer = mp.colors.Normalize()
color_map = mp.colors.LinearSegmentedColormap(
"my_map",
{
"red": [(0, 1.0, 1.0),
(1.0, .5, .5)],
"green": [(0, 0, 0),
(1.0, 0, 0)],
"blue": [(0, 0, 0),
(1.0, 0, 0)]
}
)
#Map xs to numbers:
N = len(xs)
x_nums = np.arange(1, N+1)
#Plot a bar graph:
plt.bar(
x_nums,
ys,
align="center",
color=color_map(data_normalizer(likeability_scores))
)
#Change x numbe