python中Thompson抽样试验的曲线图

我正在用python学习数据科学，最近我实现了一个基于Beta分布的Thompson采样演示。我想直观地展示Thompson采样如何快速选择最佳解决方案。为此，我在pandas中创建了一个数据框来存储“试验”信息。每行是一个试验，每列代表一个“变体”。我在数据框中用1标记随机选取的列，其他列将为零。我以如下所示的数据帧结束：

         0    1
0  1.0  0.0
1  0.0  1.0
2  1.0  0.0
3  0.0  1.0
4  1.0  0.0
5  0.0  1.0
6  0.0  1.0
7  0.0  1.0
8  0.0  1.0
9  0.0  1.0

现在，我想显示一个折线图，显示在试验过程中选择的变体。我该怎么做？我想象的是每列一个线条图（变体）。在这里，我们将在同一个图形中有两个直线图。它们都共享相同的x轴（试验）。

df.plot（）

似乎很可能。