Я пытаюсь обновить подмножество мультииндексированной серии серией, имеющей тот же индекс (см. MWE ниже). У меня есть метод, который не работает, хотя имхо должен. У меня есть медленный метод, который работает. И у меня есть быстрый метод, который работает, но выдает предупреждение.
Я хотел бы использовать быстрое построение методов на встроенных модулях, поскольку циклы отнимают много времени.
Кто-нибудь знает хороший обходной путь или правильное использование методов pandas?
import pandas as pd
j = 0
I = range(1)
J = range(2)
K = range(3)
mi_ijk = pd.MultiIndex.from_product([I, J, K], names=['i', 'j', 'k'])
mi_ik = pd.MultiIndex.from_product([I, K], names=['i', 'k'])
s0_ijk = pd.Series(0, index=mi_ijk)
s1_ik = pd.Series(1, index=mi_ik)
# does not work
s0_ijk[:, j].update(s1_ik)
print(s0_ijk.sum())
# does work, but slow
for (i, k) in s1_ik.index:
s0_ijk[i, j, k] = 1
print(s0_ijk.sum())
# does work but warning
s0_jik = s0_ijk.reorder_levels(['j', 'i', 'k'])
s0_jik[j].update(s1_ik)
s0_ijk = s0_jik.reorder_levels(['i', 'j', 'k'])
print(s0_ijk.sum())
