¿Cómo contar filas con condiciones en Pandas?

Si quieres contar filas con una condición específica en un DataFrame de pandas, puede usar el método “sum()” después de aplicar una “condición booleana” o usar el método “len()” combinado con “indexación booleana”.

Contenidos

Método 1: Usar la función sum()

El método “sum()” cuenta el número de filas en Pandas DataFrame que cumplen una condición específica. Cuando se aplica una condición booleana a una columna de DataFrame, devuelve una serie booleana con la misma longitud que DataFrame.

En esta serie, True corresponde a las filas donde se cumple la condición y False corresponde a las filas donde no se cumple la condición.

Dado que True se trata como 1 y False como 0 , el método sum() en esta serie booleana calcula la suma de todos los valores True, igual al número de filas donde se cumple la condición.

Ejemplo

import pandas as pd

# Sample DataFrame
data = {
  "A": [11, 21, 31, 41, 51],
  "B": [10, 20, 30, 40, 50]
}

df = pd.DataFrame(data)

print("Original DataFrame:")
print(df)

# Count rows where column A is greater than 21
count_rows = (df["A"] > 21).sum()

print("\nNumber of rows where column A is greater than 21:", count_rows)

Producción

Original DataFrame:
    A    B
0  11   10
1  21   20
2  31   30
3  41   40
4  51   50

Number of rows where column A is greater than 21: 3

Método 2: Usar la función len()

Puede contar filas con una condición específica en un DataFrame de pandas usando el método “len()” en combinación con “indexación booleana”.

Ejemplo

import pandas as pd

# Sample DataFrame
data = {
  "A": [11, 21, 31, 41, 51],
  "B": [10, 20, 30, 40, 50]
}

df = pd.DataFrame(data)

print("Original DataFrame:")
print(df)

# Count rows where column A is greater than 21
count_rows = len(df[df["A"] > 21])

print("\nNumber of rows where column A is greater than 21:", count_rows)

Producción

Original DataFrame:
    A   B
0  11   10
1  21   20
2  31   30
3  41   40
4  51   50

Number of rows where column A is greater than 21: 3

En este código, creamos un marco de datos de muestra con las columnas “A” y “B” .

Luego aplicamos una condición booleana ( df[“A” ] > 21 ) a la columna “A” y usamos la indexación booleana para filtrar las filas donde se cumple la condición.

Por último, use el método “len()” para contar el número de filas en el DataFrame filtrado, correspondiente al número de filas donde se cumple la condición.