python - 如何创建按列和列值过滤的函数？

Question

数据框如下所示：

index      company   body-style  wheel-base  length engine-type  \
 0  alfa-romero  convertible        88.6   168.8        dohc   
 1  alfa-romero  convertible        88.6   168.8        dohc   
 2        audi        sedan        99.4   176.6         ohc   

我一直在尝试沿着这条线使用一些东西，但遇到了错误。

def dosomething1(column,value):
    value_name=str(value)
    column_name=str(column)
    newdf=df[df[column_name]==value_name]

dosomething1(company, audi)

标签： pythondataframefilter

Answer 1

您的功能没有return声明：

它应该是：

def dosomething1(column,value):
    value_name=str(value)
    column_name=str(column)
    newdf=df[df[column_name]==value_name]
    return newdf

当你调用你的函数时，字符串应该用引号引起来，例如：

dosomething1('company', 'audi')