在同一函数中引用局部变量

Question

我想使用一个函数合并 2 个 DataFrame。 该函数在使用变量“x = 1”调用时创建 DataFrame df1，然后在使用“x！= 1”调用时创建另一个 df2，基于函数中的 if 语句 - 为进一步清楚起见，下面的代码 sn-p。

到达“df3 = pd.concat”行时，我收到错误“UnboundLocalError: local variable 'df1' referenced before assignment”。

我想了解如何实现将df1和df2连接成df3的结果。

def Concat(url, x):
    if x == 1:
        df1 = pd.read_json(url)
    else:
        df2 = pd.read_json(url)
        df3 = pd.concat([df1, df2], ignore_index=True)


def main():
    Concat('*url*', 1)
    Concat('*url*', 2)

Answer 1

你应该稍微调整一下，变成：

def Concat(url, x):
    for i in x:
        if i == 1:
            df1 = pd.read_json(url)
        else:
            df2 = pd.read_json(url)
            df3 = pd.concat([df1, df2], ignore_index=True)


def main():
    Concat('*url*', [1, 2])