Python pd с использованием переменной с именем столбца в нотации groupby - PullRequest
0 голосов
/ 01 июля 2018

Я пытаюсь использовать список, содержащий имена столбцов для моих групповых обозначений. Моя конечная цель состоит в том, чтобы пройтись по нескольким столбцам и выполнить расчет без необходимости перезаписывать одну и ту же строку несколько раз. Возможно ли это?

 a_list = list(['','BTC_','ETH_'])
 a_variable = ('{}ClosePrice'.format(a_list[0]))

 proccessing_data['RSI'] = proccessing_data.groupby('Symbol').**a_variable**.transform(lambda x: talib.RSI(x, timeperiod=14))

это ошибка, которую я сейчас получаю, потому что она думает, что я хочу столбец 'a_variable', который не существует.

AttributeError: 'DataFrameGroupBy' object has no attribute 'a_variable'

1 Ответ

0 голосов
/ 01 июля 2018

Видимо, эта запись ниже работает:

proccessing_data['RSI'] = proccessing_data.groupby('Symbol')[('{}ClosePrice'.format(a_list[0]))].transform(lambda x: talib.RSI(x, timeperiod=14))
...