python - 如何根据条件绘制图表

Question

我想在特定条件下绘制图表，而不操作我的数据框。

例如，我创建了一个带有数据框的计数图，其中有一堆小于 100 的 x 值，在计数图中，那些小于 100 的值显示为“无栏”，并且它占用了空间。所以我想去掉那些空的(计数< 100)。

我尝试创建另一个仅计数值高于 100 的数据框，但我想知道是否有更简单/更干净的方法来绘制图表，而不是创建整个数据框。

plt.figure(figsize=(10,50))
plt.ylim(100,500)
sns.countplot(data=df, x='brand')

从这段代码中，我看到许多空条是由计数值小于 100 引起的，因为 xlim 设置为 100-500。

Answer 1

import matplotlib.pyplot as plt
import seaborn as sns

plot_data = df.groupby('brand').size().reset_index(name='count').query('count>=100')
plt.figure(figsize=(10,50))
plt.ylim(100,500)
sns.barplot(data=plot_data, x='brand', y='count')