from matplotlib.collections importLineCollectionimportnumpy as npimportmathimportmatplotlib.pyplot as plt
pi= 3.1415x= np.linspace(0, 4*pi, 100)
y= [math.cos(xx) for xx inx]
lwidths=abs(x)
color=[]for i inrange(len(y)):if i < 5:
color.append('#FF0000')else:
color.append('#000000')print(x)print(y)print('--------------------------------------')
points= np.array([x, y]).T.reshape(-1, 1, 2)print(points)print('--------------------------------------')
segments= np.concatenate([points[:-1], points[1:]], axis=1)print(segments)
lc= LineCollection(segments, linewidths=lwidths, color=color)
ax=plt.axes()
ax.set_xlim(min(x), max(x))
ax.set_ylim(min(y), max(y))
ax.add_collection(lc)
plt.show()'''fig, a = plt.subplots()
a.add_collection(lc)
a.set_xlim(0, 4*pi)
a.set_ylim(-1.1, 1.1)
fig.show()'''