shadyquant
😎
This python package allows you to quantile and plot lines where you have multiple samples, typically for visualizing uncertainty. Your data should be shaped (N, T), where N is the number of samples, T is the dimension of your lines.
Install
pip install shadyquant
Usage
Consider you have 100 lines that you want to compute confidence intervals (quantiles) on:
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0,2,1000)
y = np.sin(x)
plt.plot(x,y)
I'll make 100 examples of this plot, each scaled by a random number.
w = np.random.normal(size=100)**2
traj = y + y * w[:, None]
plt.plot(traj.T, color='C0')
plt.show()
You can use shadyquant to plot a 95% confidence interval (default):
import sq
sq.traj_quantile(traj)
You can also do weighted quantiling, if you have weights attached to each line
sq.traj_quantile(traj, weights=w)
You may want to do fancy shading, which just plots a series of quantiles as polygons with transparency. The quantiles overlap, which gives a nice blending. The outer edges of the polygons still correspond to the 95% confidence interval.
sq.traj_quantile(traj, fancy_shading=True)
Here are some further options you can modify:
plt.figure(figsize=(8, 3))
ax = plt.gca()
sq.traj_quantile(
traj,
ax=ax,
fancy_shading=True,
lower_q_bound=1 / 3,
upper_q_bound=2 / 3,
levels=100,
color="red",
alpha=0.01,
)
4 Dec 2, 2022
162 Nov 11, 2022
2 Nov 29, 2021
4 Jun 12, 2022
0 Jul 9, 2022
38 Dec 20, 2022
97 Nov 4, 2022
30 Dec 10, 2022
80 Sep 17, 2022
1 Dec 30, 2021
394 Dec 18, 2022
138 Dec 14, 2022
1.6k Jan 6, 2023
156 Jan 3, 2023
502 Dec 28, 2022
76 Dec 15, 2022
338 Nov 25, 2022
102 Dec 24, 2022
70 Dec 27, 2022
31 Oct 11, 2022
2 Nov 07, 2021
2.7k Jan 07, 2023
1 Jan 07, 2022
3.2k Jan 01, 2023
14 Aug 14, 2022
3 Aug 29, 2022
4 Oct 08, 2021
6 Apr 09, 2022
2 Jan 22, 2022
9 Jul 15, 2022
73 Oct 02, 2022
145 Jan 01, 2023
3 Feb 17, 2022
1 Jan 11, 2022
2 Dec 15, 2021