基于约束从Pandas DataFrame中选择行

e7arh2l6  于 2022-11-27  发布在  其他
关注(0)|答案(1)|浏览(121)

我有几个数据集,我导入为csv文件,并显示在一个数据框Pandas。csv文件是关于新冠病毒更新的信息。
数据集有几个与此相关的列,例如“country_region”、“last_update”和“confirmed”。
比如说,我想查找德国的新冠肺炎确诊病例。
我正在尝试编写一个函数,该函数将返回与这些约束相对应的DataFrame切片,以便能够显示我正在查找的匹配项。
我需要以某种通用的方式执行此操作,以便可以提供任何列中的任何值。
我希望我有一些代码,但我坚持如何进行,我发现网上的一切只指定查找与预定义值相关的值。

vawmfj5a

vawmfj5a1#

像这样的东西?

def filter(county_region_val, last_update_val, confirmed_val, df):
    df = df.loc[((df['county_region'] == county_region_val) & (df['last_update'] == last_update_val) & (df[''confirmed'] == confirmed_val)).reset_index(drop=True)
    return df

相关问题