Как передать локальную переменную в пользовательскую функцию в пандах?

Question

Я обрабатываю dataframe pandas df. В зависимости от определенных условий в пределах df, я хочу передать df (1) либо пользовательской функции filter(), либо (2) - calculate_Result().

Код состояния является:

if df.Col1.str.contains("yes").sum() > 0: 
    df = filter_df 
    filter() # Loop 
else: 
    df = calculate_df 
    calculate_result() # End 

Однако я получая UnboundLocalError: local variable referenced before assignment.

я получаю эту ошибку и, когда я поставил df = calculate_df & df = filter_df задания в самом определении функций:

def filter(): 
    df = filter_df 
    ... 

и

def calculate_results(): 
    df = calculate_df() 
    ... 

независимо от того, где я ставлю задания, я получая UnboundLocalError: local variable referenced before assignment.

Как правильно передать локальную переменную для пользовательской функции в панд?

Answer 1

Это не связано с передачей переменной функции.

Вы использовали имя df, чтобы отнестись к глобальному модулю - в df.Col1... и в результате filter_df. Это вызывает ошибку, которую вы видите.

Используйте другое имя для локальной переменной.