python - Pandas 中样本 block 的便捷方法？

Question

我有跟踪一组人随时间变化的数据。举一个小例子，它看起来像这样:

ID    TIME   HEIGHT
0     0      10.2
0     1      3.3
0     2      2.1
1     0      11.3
1     1      8.6
1     2      9.1
2     0      10.0
2     1      35.0
2     2      4.1
.
.
.
100   0      1.0
100   1      3.0
100   2      9.0

为了便于说明，ID 指的是特定的人。因此，对于 ID=0 的所有值，在 x 轴上绘制 TIME 和在 y 轴上绘制 HEIGHT 可以得出变化以人 0 的高度计算。

我想绘制这些人的随机样本并绘制它们。例如，我想绘制 3 个人随时间的高度变化。然而，应用通常的 df.sample(3) 并不总能确保我获得特定人员的所有时间，而是会随机选择 3 行并绘制它们。 pandas 中是否有一种首选/方便的方法来对随机组进行采样？

很多这样的问题one似乎是从每个组中采样，这不是我想要做的。

Answer 1

您想要在 x 轴上绘制 'TIME'，然后获取一个以 'TIME' 作为索引、'ID'< 的矩形数据框 作为列。从那里，使用 sample 和 axis=1 对列进行采样并保持索引不变。

df.set_index(['TIME', 'ID']).HEIGHT.unstack().sample(3, axis=1).plot()