función sciPy stats.binned_statistic() | Python

stats.binned_statistic(x, values, statistic='mean', bins=10, range=None)La función calcula el valor estadístico agrupado para los datos dados (elementos de array).
Funciona de manera similar a la función de histograma . Como función de histograma, hace contenedores y cuenta el no. de puntos en cada bin; esta función calcula la suma, la media, la mediana, el recuento u otras estadísticas de los valores de cada contenedor.

Parámetros:
arr: [array_like] array de entrada que se va a agrupar.
valores: [array_like] sobre los cuales se calcularán las estadísticas.
estadísticas: estadísticas para calcular {media, recuento, mediana, suma, función}. El valor predeterminado es malo.
bin : [int o escalares] Si bins es un int, define el número de bins de igual ancho en el rango dado (10, por defecto). Si bins es una secuencia, define los bordes del bin.
rango: (flotante, flotante) Rango inferior y superior de los contenedores y, si no se proporciona, el rango es de x.max() a x.min().

Resultados: valor de estadísticas para cada contenedor; bordes de contenedores; número de contenedor

Código #1:

# stats.binned_statistic() method 
import numpy as np
from scipy import stats
   
# 1D array
arr = [20, 2, 7, 1, 34]
print("\narr : \n", arr) 
  
  
# median  
print("\nbinned_statistic for median : \n", stats.binned_statistic(
        arr, np.arange(5), statistic ='median', bins = 4)) 

Producción :

arr : 
 [20, 2, 7, 1, 34]

binned_statistic for median : 
 BinnedStatisticResult(statistic=array([ 2., nan,  0.,  4.]), 
bin_edges=array([ 1.,  9.25, 17.5, 25.75, 34.  ]), 
binnumber=array([3, 1, 1, 1, 4], dtype=int64))

 
Código #2:

# stats.binned_statistic() method 
import numpy as np
from scipy import stats
  
# mean  
arr = [20, 2, 7, 1, 34]
print("\nbinned_statistic for mean : \n", stats.binned_statistic(
        arr, np.arange(5), statistic ='mean', bins = 2)) 

Producción :

binned_statistic for mean : 
 BinnedStatisticResult(statistic=array([2., 2.]), 
bin_edges=array([ 1., 17.5, 34. ]), 
binnumber=array([2, 1, 1, 1, 2], dtype=int64))