python - sns stripplot 仅包含前 n 个类别

Question

我有可以很好地绘制 sns stripplot 的代码:

f, ax = plt.subplots(figsize=(15,12))
sns.stripplot(data = cars, x='price', y='model', jitter=.5)
plt.show()

但是汽车模型太多，所以我希望仅可视化数据集中最常出现的前 n 个汽车模型。另外，是否有任何 lambda 计算或类似的东西可以应用于 price 或 model 而无需创建单独的数据框？

如果有更好的可视化库可以提供帮助，请随时提出。

Answer 1

您可以使用 value_counts() 查找列中出现次数最多的值。在这里，我选择了最常见的 2 个模型:

most_occurring_values = cars['model'].value_counts().head(2).index

然后您可以过滤原始数据框并仅选择包含频率最高的模型的行:

cars_subset = cars[cars['model'].isin(most_occurring_values)]

最后，使用该子集来绘制数据:

f, ax = plt.subplots(figsize=(15,12))
sns.stripplot(data = cars_subset, x='price', y='model', jitter=.5)
plt.show()