python - 绘制数据框的子集而不冗余

Question

这里有一点 Python 新手。作为初学者，从培训类(class)中学习不同的函数和方法很容易，但学习如何“最好”地使用 Python 编写代码却是另一回事。

我有一个简单的场景，我希望绘制数据帧 spdf 的一部分。我只想绘制 speed 大于 0 的实例，并使用 datetime 作为 X 轴。我设法完成工作的方式对我来说似乎非常多余:

ts = pd.Series(spdf[spdf['speed']>0]['speed'].values, index=spdf[spdf['speed']>0]['datetime'])

ts.dropna().plot(title='SP1 over Time')

是否有更好的方法来绘制此数据，而无需两次指定数据帧的子集？

Answer 1

您不需要构建新系列。您可以使用原始的 df

进行绘图

df[df['col'] > 0]].plot()

就您而言:

spdf[spdf['speed'] > 0].dropna().plot(title='SP1 over Time')

我不确定您的 spdf 对象是什么或它是如何创建的。如果您经常需要使用“日期时间”列进行绘图，您可以将其设置为 df 的索引。如果您从 csv 读取数据，您可以使用 parse_dates 关键字参数，或者如果您已经有了 df，您可以使用 df.set_index('datetime') 更改索引。您可以使用 df.info() 来查看索引当前使用的内容及其数据类型。