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。

标签： pythonpandasdata-science

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')