python - 在 Pandas 中使用 GroupBy 时如何设置条件？

Question

我使用以下代码制作了 distplot。

data_agg = data.groupby('HourOfDay')['travel_time'].aggregate(np.median).reset_index()

plt.figure(figsize=(12,3))
sns.pointplot(data.HourOfDay.values, data.travel_time.values)
plt.show()

但是，我只想选择 8 小时以上而不是 0-7 小时。我该怎么做？

标签： pythonpandaspandas-groupby

Answer 1

先过滤怎么办？

data_filtered = data[data['HourOfDay'] > 7]
# depending of the type of the column of date 
data_agg = data_filtered.groupby('HourOfDay')['travel_time'].aggregate(np.median).reset_index() 
plt.figure(figsize=(12,3))
Sns.pointplot(data_filtered.HourOfDay.values, data_filtered.travel_time.values) 
plt.show()