оператор if: вернуть NaN, если число точек данных меньше, чем x, иначе рассчитать годовой доход для фрейма данных - PullRequest
0 голосов
/ 25 февраля 2019

У меня есть датафрейм, который содержит ежемесячный журнал возвратов разных фондов.Dataframe состоит из более чем 100 столбцов и 500 строк.Для дат, для которых возврат не доступен, сообщается значение np.nan.Я пытаюсь рассчитать годовой доход за последние 3 года для каждого фонда и использую следующую функцию:

def Calculate_3Y_return (data):

 data_cumsum = data.iloc[-36:].cumsum()
 return_3Y = data_cumsum.iloc[-1]*(1/3)

 return_3Y

Для средств, которые не имеютДостаточно точек данных (их послужной список короче, чем 36 месяцев). Я бы хотел, чтобы моя функция возвращала NaN.

Можете ли вы помочь мне добавить оператор if в мою функцию, который будет рассчитывать годовой доход за 3 года для средств, которые имеют36 точек данных доступны и возвращают NaN для фондов, которые имеют менее 36 точек данных (эти фонды были запущены недавно, и поэтому количество доступных точек данных меньше 36).

Я уже пытался использовать dataframe.count () но это не помогло

Большое спасибо

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...