Pandas:使用set_categories时获取SettingWithCopyWarning

时间:2016-06-15 15:14:29

标签: python pandas dataframe categorical-data chained-assignment

我有两个数据框。两者都具有相同的列集,但是某些列是分类类型的(基于实际包含的值)。为了组合它们,我使用两个值的并集刷新分类列的分类类型。

def appendDFsWithCat(df1, df2):
    columns = df1.select_dtypes(include=['category']).columns
    for c in columns:
        catValues1 = list(df1[c].cat.categories)
        catValues2 = list(df2[c].cat.categories)
        catValues = list(set(catValues1 + catValues2))
        df1[c] = df1[c].cat.set_categories(catValues)
        df2[c] = df2[c].cat.set_categories(catValues)
    return df1.append(df2, ignore_index=True).reset_index(drop=True)

一切都像预期的那样工作但我想理解为什么在执行此代码时SettingWithCopyWarning会引发:

df1[c] = df1[c].cat.set_categories(catValues)
Utility.py:149: SettingWithCopyWarning:

我发现没有其他可能刷新类别数据而不是使用过的。

1 个答案:

答案 0 :(得分:1)

这很可能是因为您传递给函数的对象。

如果我设置以下示例:

cats1 = pd.Series(['a', 'a', 'b', 'b'], name='cat', dtype="category")
data1 = pd.Series([1, 2, 3, 4], name='val', dtype=np.int64)
df1 = pd.concat([cats1, data1], axis=1)

并运行你的功能:

print appendDFsWithCat(df1, df1)

我没有错误和输出:

  cat  val
0   a    1
1   a    2
2   b    3
3   b    4
4   a    1
5   a    2
6   b    3
7   b    4

但是,如果我运行它:

print appendDFsWithCat(df1.iloc[:-1], df1)

我收到以下警告:

C:\Anaconda2\lib\site-packages\ipykernel\__main__.py:7: SettingWithCopyWarning: 
A value is trying to be set on a copy of a slice from a DataFrame.
Try using .loc[row_indexer,col_indexer] = value instead

See the caveats in the documentation: http://pandas.pydata.org/pandas-docs/stable/indexing.html#indexing-view-versus-copy

这个输出:

  cat  val
0   a    1
1   a    2
2   b    3
3   a    1
4   a    2
5   b    3
6   b    4

如果您阅读警告,它会告诉您正在尝试在对象上设置值,该对象是另一个对象的切片或视图。这意味着您在特定位置分配值的数据帧仅是对另一个对象的引用。我通过将数据帧传递给我知道是切片或视图的函数来制造这种情况。

你可以通过强迫对象成为他们自己的东西来解决这个问题:

解决方案

def appendDFsWithCat(df1, df2):
    # I added this line to ensure they are their own dataframes
    df1, df2 = df1.copy(), df2.copy()
    columns = df1.select_dtypes(include=['category']).columns
    for c in columns:
        catValues1 = list(df1[c].cat.categories)
        catValues2 = list(df2[c].cat.categories)
        catValues = list(set(catValues1 + catValues2))
        df1[c] = df1[c].cat.set_categories(catValues)
        df2[c] = df2[c].cat.set_categories(catValues)
    return df1.append(df2, ignore_index=True).reset_index(drop=True)

现在我跑的时候:

print appendDFsWithCat(df1.iloc[:-1], df1)

我明白了:

  cat  val
0   a    1
1   a    2
2   b    3
3   a    1
4   a    2
5   b    3
6   b    4

现在有警告。