为何筛选出来不是所想的 | python | python 技术论坛-大发黄金版app下载

答案

grouped = df.groupby(['年龄', '学校'])
# 方法一：不使用 `in`
filtered = grouped.filter(lambda x: len(x) > 1 and (x.类别 != 1).all())
print(filtered)
# 方法二：使用`in`前，先将 `x.类别` 转换为列表
filtered = grouped.filter(lambda x: len(x) > 1 & 1 not in list(x.类别))
print(filtered)

分析

参考这里：

这是因为 x.类别 返回的是 pandas.series 对象，对其进行 in 操作会被解释为对series.__contains__(value)的调用。

而参考 pandas.generic.py 源码，实际操作的是_info_axis 属性

def __contains__(self, key) -> bool_t:
        """true if the key is in the info axis"""
        return key in self._info_axis

而对于pandas.series 对象来说，_info_axis 属性用来表示其行索引。 可以用下面的代码来测试：

# 打印每个分组的数据
for name, x in grouped:
    # 分组字段，元组
    print(name)
    # series对象的 _info_axis，这里实际上是源数据的行索引:
    print(x.category._info_axis)
    # 判断1是否在行索引组成的列表中：
    print(1 not in x.category._info_axis)
    # 再加上分组长度的判断：
    print(len(x) > 1 & 1 not in x.category._info_axis)
# 按需求: 将分组后的数据大于1条 且 category 全不等于1 的数据筛选出来，上面显然不符合。因此不能直接使用 `1 not in x.category`

所以这里直接用 in 会出现这种现象。

还是改为英文变量名，方便输入，下面是完整代码：

import pandas as pd
df = pd.dataframe([['小明', 15, '一中', 99, 4],
                   ['小美', 12, '一中', 63, 1],
                   ['灰灰', 19, '一中', 85, 2],
                   ['铭铭', 12, '一中', 76, 2],
                   ['豪豪', 15, '一中', 55, 4],
                   ['小黄', 18, '二中', 68, 1],
                   ['小黑', 18, '二中', 85, 1]],
                  columns=['name', 'age', 'school', 'score', 'category'])
# 打印出数据，便于后面分析
# print(df)
# 对 age 和 school 分组
grouped = df.groupby(['age', 'school'])
# 打印每个分组的数据
for name, x in grouped:
    # 分组字段，元组
    print(name)
    # series对象的 _info_axis，这里实际上是源数据的行索引:
    print(x.category._info_axis)
    # 判断1是否在行索引组成的列表中：
    print(1 not in x.category._info_axis)
    # 再加上分组长度的判断：
    print(len(x) > 1 & 1 not in x.category._info_axis)
# 按需求: 将分组后的数据大于1条 且 category 全不等于1 的数据筛选出来，上面显然不符合。因此不能直接使用 `1 not in x.category`
grouped = df.groupby(['age', 'school'])
# 方法一：不使用 `in`
filtered = grouped.filter(lambda x: len(x) > 1 and (x.category != 1).all())
print(filtered)
# 方法二：使用`in`前，先将 `x.category` 转换为列表
filtered = grouped.filter(lambda x: len(x) > 1 & 1 not in list(x.category))
print(filtered)

非常感谢

df1=( df.groupby(['年龄','学校']) .filter (lambda x:(len (x)>1) & (1 not in x.类别.values)) )