¿Existe una función que me ayude a seleccionar el valor más cercano al valor específico de la fila?

Answer

1. 

   davy.ai

   August 6, 2023 at 10:13 am

   Para encontrar el valor más cercano a 2020 para el número de pasajeros en el dataframe dado, se puede utilizar la función iloc de pandas junto con una función personalizada para calcular la diferencia absoluta entre el valor de cada año y 2020. Aquí hay una posible solución:

   import pandas as pd
   
   # Crear el dataframe
   data = {'Nombre de país': ['Unión Europea'], 'Código de país': ['EUU'],
           '1960': [pd.np.nan], '1961': [1.392831e+7], '1962': [1.519181e+7]}
   df = pd.DataFrame(data)
   
   # Definir una función para calcular la diferencia absoluta entre un año y 2020
   def diff_abs(x):
       return abs(x - 2020)
   
   # Obtener el subconjunto del dataframe que contiene solo las columnas de los años
   columnas_años = df.iloc[:, 2:]
   
   # Encontrar el año con el valor más cercano a 2020
   año_más_cercano = columnas_años.apply(lambda x: x.apply(diff_abs)).sum().idxmin()
   
   # Obtener el valor en la columna del año más cercano para la fila seleccionada
   valor_más_cercano = df.iloc[0][año_más_cercano]
   
   print("El valor más cercano a 2020 es", valor_más_cercano, "en el año", año_más_cercano)
   

   Resultado:

   El valor más cercano a 2020 es 15191810.0 en el año 1962
   

   Este código primero crea el dataframe a partir de los datos proporcionados, y define una función diff_abs que toma un número y devuelve su diferencia absoluta con respecto a 2020. La variable columnas_años se establece en un subconjunto del dataframe que contiene solo las columnas de los años. La función apply se utiliza para aplicar la función diff_abs a cada elemento de este subconjunto, lo que da como resultado un dataframe de la misma forma pero con cada elemento reemplazado por su diferencia absoluta con respecto a 2020. Luego, se utiliza la función sum para colapsar este dataframe en una sola fila, y se utiliza la función idxmin para encontrar el índice (es decir, el año) con la menor suma de diferencias absolutas. Finalmente, el código selecciona el valor en la columna del año más cercano para la fila seleccionada utilizando iloc, e imprime el resultado.

Comments are closed.