Индексирование мультииндексированного фрейма данных с использованием списка

У меня есть мультииндексированный фрейм данных pandas, созданный так:

m_index = ['time_remaining', 'inventory']
for i in indicators:
    m_index.append(i.name)
cols = []
for col in m_index:
    cols.append(col)
cols.append('action')
cols.append('cost')

optimal_actions = pd.DataFrame(columns=cols)
optimal_actions.set_index(m_index, inplace=True)

Затем у меня есть список значений для индекса:

state_variables = [indicator.value for indicator in indicators]
state = [time_remaining, i*trade_size]      
state.extend(state_variables)

И я пытаюсь установить строку, индексированную в списке значений, к значению:

state = [300, 1, 1.0]
optimal_actions.loc[state] = [-1, -0.2]

Это дает мне неприятную ошибку:

ValueError: zero-size array to reduction operation maximum which has no identity

Любая помощь приветствуется!

1 ответ

Решение

Если кому-то это нужно, я решил решить эту проблему путем индексации мультииндекса по кортежу, а не по списку. Таким образом, модификация приведенного выше кода будет:

state = [300, 1, 1.0]
optimal_actions.loc[tuple(state)] = [-1, -0.2]
Другие вопросы по тегам