Dado un número, cómo conseguir que el anterior y el siguiente valor dentro de una lista? [cerrado]

0

Pregunta

Tengo el siguiente dataframe:

VALUE_TO_FIND  UNORDERED_LIST
5              [0,10]
3              [1,0,10,8,4,2]
2              [9,10,0]
5              [4,8,0,1,2,10]
4              [0,10,4]

Dado el valor de la columna VALUE_TO_FINDcómo puedo conseguir el anterior y el siguiente valor en el interior de UNORDERED_LIST?

VALUE_TO_FIND  UNORDERED_LIST   PREVIOUS_VALUE   NEXT_VALUE
5              [0,10]           0                10
3              [1,0,10,8,4,2]   2                4
2              [9,10,0]         0                9
5              [4,8,0,1,2,10]   4                8
4              [0,10,4]         4                4

Nota: UNORDERED_LIST siempre tendrá el máximo y el mínimo de VALUE_TO_FIND

dataframe pandas python
2021-11-24 02:55:34
1

Mejor respuesta

1

Podría ser una solución más eficiente, pero he tratado de zipping las dos columnas relevantes:

import pandas as pd
df = pd.DataFrame({'VALUE_TO_FIND': [5, 3, 2, 5, 4,],
    'UNORDERED_LIST': [[0,10], [1,0,10,8,4,2], [9,10,0], [4,8,0,1,2,10], [0,10,4],]})

def find_prev_next(x, lst):
    return max(y for y in lst if y <= x), min(y for y in lst if y >= x) # tuple

_ = [find_prev_next(x, lst) for x, lst in zip(df.VALUE_TO_FIND, df.UNORDERED_LIST)]
df['PREVIOUS_VALUE'], df['NEXT_VALUE'] = zip(*_)

print(df)
   # VALUE_TO_FIND       UNORDERED_LIST  PREVIOUS_VALUE  NEXT_VALUE
# 0              5              [0, 10]               0          10
# 1              3  [1, 0, 10, 8, 4, 2]               2           4
# 2              2           [9, 10, 0]               0           9
# 3              5  [4, 8, 0, 1, 2, 10]               4           8
# 4              4           [0, 10, 4]               4           4
2021-11-24 03:32:02

En otros idiomas

Esta página está en otros idiomas

Русский
..................................................................................................................
Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
中文
..................................................................................................................
Slovenský
..................................................................................................................