如何从主DataFrame拆分出模态数≤2及剩余列的两个子DataFrame？

错误原因分析 #

• 尝试1报错TypeError: 'method' object is not subscriptable：df.nunique是Pandas的方法，必须加括号调用（df.nunique()）才能获取各列的唯一值数量，直接用df.nunique[0]是对方法对象做下标操作，必然报错。另外iloc是按位置索引，这里更适合用列名筛选。
• 尝试2未实现筛选：同样没调用df.nunique()方法，且逻辑偏离需求——你要筛选唯一值≤2的列，但sort_values()[:11]只是取排序后的前11列，同时代码末尾多了一个冗余括号)。

正确实现步骤 #

1. 获取各列的唯一值数量

nunique_counts = df.nunique()

2. 筛选唯一值数量≤2的列名

low_cardinality_cols = nunique_counts[nunique_counts <= 2].index

3. 生成两个子DataFrame

# 第一个DataFrame：包含所有唯一值≤2的列
df_low = df[low_cardinality_cols]

# 第二个DataFrame：包含剩余所有列
df_high = df.drop(low_cardinality_cols, axis=1)

简化写法 #

也可以一步完成，不需要单独存储列名：

df_low = df.loc[:, df.nunique() <= 2]
df_high = df.drop(df_low.columns, axis=1)

内容的提问来源于stack exchange，提问作者MlleBlinie