Булева маска в графстве Seaborn

Question

Я хочу применить эту логическую маску

csv["country"].value_counts()>5000

к этой функции

sns.countplot(y = csv["country"].value_counts()>5000, data = csv)

, но возникает эта ошибка:

"Unalignable boolean Series provided as indexer (index of the boolean Series and of the indexed object do not match)".

Как мне поступить

Answer 1

Вы можете сделать что-то вроде:

s = csv['country'].value_counts()

s[s > 5000].plot(kind='bar')

Чтобы использовать seaborn, вы можете отфильтровать данные, используя:

s = csv['country'].value_counts()
s = s[s > 5000].index.tolist()

sns.countplot(x='country', data=csv.query("country in @s")) # option1
# sns.countplot(x='country', data=csv.loc[df["country"].isin(s))) # option2