gpt4 book ai didi

python - 具有 lambda 函数的 Pandas .filter() 方法

转载 作者:行者123 更新时间:2023-11-28 19:04:44 24 4
gpt4 key购买 nike

<分区>

我正在尝试理解 .filter() Pandas 中的方法。我不确定为什么下面的代码不起作用:

# Load data
from sklearn.datasets import load_iris
import pandas as pd
data = load_iris()
df = pd.DataFrame(data.data, columns=data.feature_names)

# Set arbitrary index (is this needed?) and try filtering:
indexed_df = df.copy().set_index('sepal width (cm)')
test = indexed_df.filter(lambda x: x['petal length (cm)'] > 1.4)

我得到:

TypeError: 'function' object is not iterable

我很欣赏有更简单的方法来做到这一点(例如 bool 索引),但出于学习目的,我试图理解为什么 filter 在适用于 groupby 时会在这里失败> 如下图:

这个有效:

 filtered_df = df.groupby('petal width (cm)').filter(lambda x: x['sepal width (cm)'].sum() > 50)

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com