Pandas функция с 2 аргументами для определения порога

Question

Мне нужно найти людей с большим или равным пороговому усилению => фрейм данных содержит столбец capitalGain с разными значениями 10,20,30,50,1000,5000,10000 ... и т. Д. Я пытаюсь:

Функция:

def get_num_people_with_higher_gain(dataframe, threshold_gain) 
    threshold_gain = dataframe["capitalGain"][dataframe["capitalGain"] >= threshold_gain].count()
        return threshold_gain

Вызов функции

df = get_num_people_with_higher_gain(dataframe, threshold_gain)

Но я получаю следующее сообщение об ошибке:

NameError                                 Traceback (most recent call last)
<ipython-input-50-5485c90412c8> in <module>
----> 1 df = get_num_people_with_higher_gain(dataframe, threshold_gain)
      2 threshold = get_num_people_with_higher_gain(dataframe, threshold_gain)

NameError: name 'dataframe' is not defined

Поскольку в функции 2 аргумента ( dataframe, threshold_gain), означает ли это, что оба должны быть как-то определены в функции?

Спасибо

Answer 1

Наконец, вот решение

def get_num_people_with_higher_gain(dataframe, threshold_gain):
    result = len(dataframe[dataframe["capitalGain"] >= threshold_gain])
    return result
result = get_num_people_with_higher_gain(dataframe,60000)
result