为什么在Pandas中使用“==”返回一个序列而不是bool？

Question

我就是不明白第二行“==”是什么意思：
-这不是测试，没有if语句…
-它不是一个变量声明

我以前从未见过这个，事情是data.ctage==cat是一个熊猫系列，而不是一个测试

for cat in data["categ"].unique():
    subset = data[data.categ == cat] # Création du sous-échantillon
    print("-"*20)
    print('Catégorie : ' + cat)
    print("moyenne:\n",subset['montant'].mean())
    print("mediane:\n",subset['montant'].median())
    print("mode:\n",subset['montant'].mode())
    print("VAR:\n",subset['montant'].var())
    print("EC:\n",subset['montant'].std())
    plt.figure(figsize=(5,5))
    subset["montant"].hist(bins=30) # Crée l'histogramme
    plt.show() # Affiche l'histogramme

Answer 1

它创建一个带有索引的布尔序列，其中data.categ等于cat，使用此布尔掩码，您可以过滤数据帧，换句话说subset将具有所有记录，其中categ是存储在cat中的值

这是一个使用数字数据的示例

np.random.seed(0)
a = np.random.choice(np.arange(2), 5)
b = np.random.choice(np.arange(2), 5)
df = pd.DataFrame(dict(a = a, b = b))


df[df.a == 0].head()

#   a   b
# 0 0   0
# 2 0   0
# 4 0   1

df[df.a == df.b].head()

#   a   b
# 0 0   0
# 2 0   0
# 3 1   1