Pandas fillna在multiindex系列

Question

具有NA的series_A由MultiIndex (X, Y)索引，而要填充的值在Series_B ，其由X索引。 如何有效地解决这类问题？

例如，这是series_A ：

bar  one    0.299368
     two         NaN
baz  one   -0.863838
     two   -0.251905
foo  one    1.063327
     two         NaN
qux  one    0.206053
     two    0.408204

和series_B包含要填写的值：

bar  0.123
foo  0.456

Answer 1

方法1
使用unstack将第一级索引放入列中，然后使用fillna

series_A.unstack(0).fillna(series_B).unstack().dropna()

bar  one    0.299368
     two    0.123000
baz  one   -0.863838
     two   -0.251905
foo  one    1.063327
     two    0.456000
qux  one    0.206053
quz  two    0.408204
dtype: float64

方法2
使用add方法来利用其level和fill_value参数，然后使用combine_first

series_A.combine_first(series_A.add(series_B, level=0, fill_value=0))

bar  one    0.299368
     two    0.123000
baz  one   -0.863838
     two   -0.251905
foo  one    1.063327
     two    0.456000
qux  one    0.206053
quz  two    0.408204
dtype: float64

方法3
使用map上的Index对象从返回series_A.index.get_level_vaues(0)与所述可调用series_B.get

series_A.fillna(
    pd.Series(series_A.index.get_level_values(0).map(series_B.get), series_A.index)
)

bar  one    0.299368
     two    0.123000
baz  one   -0.863838
     two   -0.251905
foo  one    1.063327
     two    0.456000
qux  one    0.206053
quz  two    0.408204
dtype: float64

方法4
使用np.isnan和np.flatnonzero来查找np.nan的位置。 然后找到要使用get_level_values和map插入的值。 最后，使用iloc放置到位

i = np.flatnonzero(np.isnan(series_A.values))
series_A.iloc[i] = series_A.index.get_level_values(0)[i].map(series_B.get)
series_A

bar  one    0.299368
     two    0.123000
baz  one   -0.863838
     two   -0.251905
foo  one    1.063327
     two    0.456000
qux  one    0.206053
quz  two    0.408204
dtype: float64