如何使python函数查询/过滤/保存数据帧

Question

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

query_function（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)调用它，它将在当前路径中创建名为dataframe.csv的文件。