在 python 中使用长格式数据创建时间序列图？

Question

我在 pandas DataFrame 中有时间序列数据，如下所示：

ID  HeartRate
1      120
1      118
1      115
2      98
2      110
2      112
3      128
3      115
3      90

我想为每个不同的 ID（即患者）创建一个单独的线图。我怎样才能最好地使用matplotlib呢？我需要创建一个“时间间隔”变量吗？

---

df = my_data[['ID', 'HR']].copy() ## creating a new "mini" dataframe from the larger one that I've got. 

n_ids = df.ID.unique().size 
n_cols = int(n_ids ** 0.5) 
n_rows = int(n_ids + n_ids % n_cols) 
fig, axes = plt.subplots(n_rows, n_cols) 
for i, (ids, hr) in enumerate(df.groupby('ID')['HR']): 
hr.plot(ax=axes[i], title=f"ID:{idx}") 
fig.tight_layout()

但是，当我收到以下错误时：

'numpy.ndarray' object has no attribute 'get_figure'

Answer 1

只需 groupby 并绘制它：

df.groupby('ID')['HeartRate'].plot()

或者使用多个轴，而不用担心（至少这么多）类别的大小：

n_ids = df.ID.unique().size
n_cols = int(n_ids ** 0.5)
n_rows = n_cols + (1 if n_ids % n_cols else 0)                   
fig, axes = plt.subplots(n_rows, n_cols)
axes = axes.ravel()
for i, (idx, series) in enumerate(df.groupby('ID')['HeartRate']):
    series.plot(ax=axes[i], title=f"ID:{idx}")
fig.tight_layout()

输出：