创建具有未指定返回次数的函数

Question

我有一个简单的代码，它采用三个数据帧（Pandas 库）并将“PROLIFIC_PID”列的值分配为该数据帧的索引

def pid_set(a, b, c): #accepts three dataframes at a time and returns dataframes with same name with PID as index
    a = a.set_index('PROLIFIC_PID')
    b = b.set_index('PROLIFIC_PID')
    c = c.set_index('PROLIFIC_PID')
    return a, b, c

我想更改上面的代码，以便它需要 X 个数据帧，将 PID 设置为索引，并返回 X 个新数据帧。 我尝试了上面代码的多种变体，但没有成功——尝试将索引数据帧放在一个列表中并只返回列表，我尝试“通过”作为返回等——但它们并没有像我希望的那样运行。

Answer 1

你可以让 X 成为一个数据框列表，比如

X = [a, b, c, ...]

def pid_set(X):
    for i in range(len(X)):
        X[i] = X[i].set_index("PROLIFIC_PID")
    return X

顺便说一句，您不必返回这些数据帧，它们甚至会在pid_set函数之外进行更新。