如何将pandas数据帧切割为函数中的参数？

Question

我要做的是将pandas数据帧切片的规则放在函数中。

例如：

row1 = {'a':5,'b':6,'c':7,'d':'A'}
row2 = {'a':8,'b':9,'c':10,'d':'B'}
row3 = {'a':11,'b':12,'c':13,'d':'C'}
df = pd.DataFrame([row1,row2,row3])

我正在以这种方式切割数据框：

print df.loc[df['a']==5]
print df.loc[df['b']==12]
print df.loc[(df['b']==12) | df['d'].isin(['A','C']),'d']

出于我的目的，我需要以不同的方式将相同的数据帧切片为函数的一部分。例如：

def slicing(locationargument):
    df.loc(locationargument)
    do some stuff..
    return something

或者，我期待getattr（）工作，但这告诉我DataFrames没有.loc [...]属性。例如：

getattr(df,"loc[df['a']==5]")

返回：

AttributeError: 'DataFrame' object has no attribute 'loc[df['a']==5]'

我在这里遗漏了什么吗？任何想法或替代方案将不胜感激！

Answer 1

在Pandas中，我认为将.loc视为DataFrame上的一个函数（或方法）是不对的。例如，语法df.loc(...)不正确。相反，您需要编写df.loc[...]（括号，而不是括号）。

那么简单地说：

def slicing(locationargument):
    df.loc[locationargument]
    do some stuff..
    return something

然后问题就变成了“locationargument应该是什么类型的对象？如果它是一个可迭代的，其长度等于数据框中的行数，那么你们都已经设置了。另一种选择可能是把它变成一个字符串然后写：

def slicing(locationargumentstring):
    df.loc[eval(locationargumentstring)]
    do some stuff..
    return something

如果您转到getattr路线，请记住该属性不包含参数。所以以下是坏事：

getattr(df, "loc[df['a']==5]")

但以下情况可行：

getattr(df, "loc")[eval("df['a']==5")]

，更直接的是，

getattr(df, "loc")[df['a']==5]