如何创建一个Python函数来查询/过滤/保存数据框

Question

数据帧有10000行和10列（string，int，float）。尝试编写一个python函数来查询、过滤并将输出csv保存在本地磁盘的当前路径中。函数如下所示：

查询函数（dataframe，column1=None，column2=None，column3=None…）

函数中的所有列都是可选的

我用谷歌搜索了一下这个话题，但没有找到真正有用的东西。有人能帮忙吗？你知道吗

Answer 1

def query_function(dataframe, **kwargs):
    for key, value in kwargs.items():
        dataframe = dataframe[(df[key] == value)]
    dataframe.to_csv('dataframe.csv')

用：query_function(dataframe, columnx=value, columny=value)调用它，它将创建一个名为数据帧.csv在当前路径中。你知道吗