使用Python中缀语法从一个函数到另一个函数的“管道”输出

Question

我试图使用Python/Pandas（作为一个学习练习）大致复制R中的dplyr包。我被困在“管道”功能上。

在R/dplyr中，这是使用管道运算符%>%完成的，其中x %>% f(y)相当于f(x, y)。如果可能的话，我想使用中缀语法复制它（请参见here）。

为了说明这一点，请考虑下面的两个函数。

import pandas as pd

def select(df, *args):
    cols = [x for x in args]
    df = df[cols]
    return df

def rename(df, **kwargs):
    for name, value in kwargs.items():
        df = df.rename(columns={'%s' % name: '%s' % value})
    return df

第一个函数接受一个dataframe并只返回给定的列。第二个获取数据帧，并重命名给定的列。例如：

d = {'one' : [1., 2., 3., 4., 4.],
     'two' : [4., 3., 2., 1., 3.]}

df = pd.DataFrame(d)

# Keep only the 'one' column.
df = select(df, 'one')

# Rename the 'one' column to 'new_one'.
df = rename(df, one = 'new_one')

为了使用pipe/infix语法实现相同的功能，代码如下：

df = df | select('one') \
        | rename(one = 'new_one')

所以|左侧的输出作为第一个参数传递给右侧的函数。每当我看到这样的事情完成时（例如，here），它就涉及lambda函数。是否可以用同样的方式在函数之间传递Pandas的数据帧？

我知道Pandas有.pipe方法，但对我来说重要的是我提供的示例的语法。任何帮助都将不胜感激。

Answer 1

您可以使用sspipe库，并使用以下语法：

from sspipe import p
df = df | p(select, 'one') \
        | p(rename, one = 'new_one')