连接多个数据框时保留原始分类映射

我托尔梅克

我有n每个只有一列的数据帧,其中有dtype category不同框架的列具有部分重叠。我想在将列连接到新框架后共享列之间的类别-> ID映射:

In [362]: af = pd.DataFrame(pd.Series(['a1','a2', 'b1'], dtype="category", name='a'))

In [365]: bf = pd.DataFrame(pd.Series(['b1','b2', 'a1'], dtype="category", name='b'))

In [373]: all_categories = pd.Categorical(['a1','a2','b1','b2'])

In [376]: show_mapping(af.a)
[('a1', 0), ('a2', 1), ('b1', 2)]

In [377]: af.a.cat =  all_categories

In [378]: show_mapping(af.a)
[('a1', 0), ('a2', 1), ('b1', 2), ('b2', 3)]

In [379]: show_mapping(bf.b)
[('b2', 0), ('a1', 1), ('b1', 2)]

In [380]: bf.b.cat =  all_categories

In [381]: show_mapping(bf.b)
[('a1', 0), ('a2', 1), ('b1', 2), ('b2', 3)]

现在,我将这些框架与假定为相同的映射一起加入:

In [382]: df = af.join(bf)

但是当我打印列的映射时,它们会重置:

In [384]: show_mapping(df.a)
[('a1', 0), ('a2', 1), ('b1', 2)]

In [385]: show_mapping(df.b)
[('b2', 0), ('a1', 1), ('b1', 2)]

为什么在加入框架后会自动重做映射,我如何实现我的预期?


def show_mapping(x):
    print(list(sorted(zip(x.cat.categories, x.cat.codes), key=lambda x : x[1])))

编辑

所有的困惑都来自使用不良表演功能。下面的show函数实际上可以正常工作。将其与set_categories方法结合,而不是.cat = ...一切都正确:

def show_mapping(s):
    print([(e, s.cat.codes[i]) for i, e in enumerate(s)])
耶斯列尔

您可以使用set_categoriesbyall_categories在每个类别列中使用相同的类别,因此在join获得相同的代码后:

af = pd.DataFrame(pd.Series(['a1','a2', 'b1'], dtype="category", name='a'))
bf = pd.DataFrame(pd.Series(['b1','b2', 'a1'], dtype="category", name='b'))

def show_mapping(x): 
    return (list(sorted(zip(x.cat.categories, x.cat.codes), key=lambda x : x[1]))) 

print(af.a)
0    a1
1    a2
2    b1
Name: a, dtype: category
Categories (3, object): [a1, a2, b1]

all_categories = pd.Categorical(['a1','a2','b1','b2'])

af.a = af.a.cat.set_categories(all_categories)
bf.b = bf.b.cat.set_categories(all_categories)

添加类别b2

print(af.a)
0    a1
1    a2
2    b1
Name: a, dtype: category
Categories (4, object): [a1, a2, b1, b2]

print(show_mapping(af.a))
[('a1', 0), ('a2', 1), ('b1', 2)]

print(show_mapping(bf.b))
[('b1', 0), ('a1', 2), ('a2', 3)]

df = af.join(bf)

print(show_mapping(df.a))
[('a1', 0), ('a2', 1), ('b1', 2)]

print(show_mapping(df.b))
[('b1', 0), ('a1', 2), ('a2', 3)]

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章