gpt4 book ai didi

python - 使用字典灵活选择 pandas 数据框行

转载 作者:太空宇宙 更新时间:2023-11-04 02:09:51 24 4
gpt4 key购买 nike

<分区>

假设我有以下数据框:

df = pd.DataFrame({'color':['red', 'green', 'blue'], 'brand':['Ford','fiat', 'opel'], 'year':[2016,2016,2017]})

brand color year
0 Ford red 2016
1 fiat green 2016
2 opel blue 2017

我知道要使用多列进行选择,我可以这样做:

new_df = df[(df['color']=='red')&(df['year']==2016)]

现在我想做的是找到一种方法来使用字典来选择我想要的行,其中字典的键代表映射到允许值的列。例如,在 df 上应用以下字典 {'color':'red', 'year':2016} 会产生与 new_df 相同的结果。

我已经可以使用 for 循环来完成它,但我想知道是否有任何更快 和/或更多“pythonic”的方式来完成它!

请包括方法所用的时间。

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