如何用條件繪制圖

Question

我想在不操縱數據框的情況下以一定條件繪制圖形。

例如，我創建了一個帶有一個數據框的計數圖，該數據框具有一堆小於100的x值，而在計數圖中，小於100的x值顯示為“ no-bar”，並且占用了空間。 所以我只想擺脫那些空的（計數<100）。

我試圖創建另一個僅具有大於100的計數值的數據框，但是我想知道是否有一種更簡單/更干凈的方法來繪制圖形，而不是創建整個數據框。

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

從此代碼中，我看到由於xlim設置為100-500，導致計數值小於100導致許多空條。

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