计数熊猫并将输出分配给变量

Question

I'm fairly new to python and pandas so forgive me if this is a somewhat basic question. I am reading in some data from a csv file, I want to do a tally from column 'gender' of 'M', 'F' and NaN. The code below outputs this:

    import pandas as pd
    import numpy as np

    df = pd.read_csv("....csv")
    count = pd.value_counts(df['gender'],dropna=False)

This outputs:

    M      22
    F       3
    NaN     1

However, I don't want to just see these as a tally, I want the values to be assigned to variables. Ie have

    male = pd.value_counts(df['gender'],'M',dropna=False)

or something similar, giving male = 22 (and the same for female and Nan), however I can't find an obvious way to do this using pandas. Any advice? Many thanks in advance!

Answer 1

In this example we take the count of the gender series filtered by == "male"

import pandas as pd
import random
df = pd.DataFrame({'gender': [random.choice(['male', 'female']) for x in range(100)]})
count_men = df[df["gender"] == "male"].count()
count_men

And if you just want the integer you can take it as the zeroth value:

count_men = df[df["gender"] == "male"].count()[0]