分类掩蔽

Question

我有一个我认为很简单的问题，但是我在文件里找不到我想要的东西。你知道吗

我有一个数据帧，它有一个Categorical列，名为mycol，类别为a和b，并希望按如下方式屏蔽数据帧的一个子集：

df_a = df[df.mycol.equal('a')]

目前我正在做：

df_a = df[df.mycol.cat.codes.values==df.mycol.cat.categories.to_list().index('a')]

这显然是非常冗长和不雅。因为df.mycol既有代码又有编码标签，所以它拥有执行此操作的所有信息，所以我想知道执行此操作的最佳方法。。。你知道吗

Answer 1

df_a = df[df["mycol"]=='a'] 

我相信这应该行得通，除非你所说的“掩码”是指你想把那些没有有效值的值归零