时间戳

首页 > 解决方案 > 带有MultiIndex DataFrame的fillna的Pandas SettingWithCopyWarning

python - 带有MultiIndex DataFrame的fillna的Pandas SettingWithCopyWarning

问题描述

带有的行fillna()会引发警告，即使它没有就地执行。这是为什么？

import pandas as pd
import numpy as np


tuples = [('foo', 1), ('foo', 2), ('bar', 1), ('bar', 2)]
index = pd.MultiIndex.from_tuples(tuples)

df = pd.DataFrame(np.random.randn(2, 4), columns=index)
df.loc[0, ('foo', 1)] = np.nan

# this works without warning
# df = pd.DataFrame({'foo': [1, np.nan, 3], 'bar': [np.nan, 22, 33]]})  

df1 = df[['foo', 'bar']]
# df1 = df[['foo', 'bar']].copy()  # this does not help
filled = df1.fillna({'foo': 100, 'bar': 200}, inplace=False)

foo如果和bar是普通列，而不是多索引，则不会出现问题。

标签： pythonpandasdataframewarningsnan

解决方案

这是一个误报，不应该在这里提出警告。我认为问题在于fillna不理解“foo”和“bar”适用于 MultiIndex 列的特定级别。

我建议在实现此功能之前调用fillna内部GroupBy作为解决方法。

fill = {'foo': 100, 'bar': 200}
df1.groupby(level=0, axis=1).apply(lambda x: x.fillna(fill[x.name]))

          foo                 bar          
            1         2         1         2
0  100.000000  1.040531 -1.516983 -0.866276
1   -0.055035 -0.107310  1.365467 -0.097696

或者，要fillna直接使用，请指定元组的字典（因为，MultiIndex），

df1.fillna({('foo', 1): 100, ('foo', 2): 100})

          foo                 bar          
            1         2         1         2
0  100.000000  1.040531 -1.516983 -0.866276
1   -0.055035 -0.107310  1.365467 -0.097696

推荐阅读