創建具有未指定返回次數的函數

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函數之外進行更新。