Python将数组从一个函数传递到另一个函数

不需要重新运行 SQL_Country() 第二次作为 df

df_country = None

def SQL_Country():
    if df_country is None:
        conn = psq.connect("localhost","root","root","world",cursorclass=psq.cursors.DictCursor)
        query = "SELECT * FROM country"

        with conn:
            cursor = conn.cursor()
            cursor.execute(query)
            country = cursor.fetchall()

            df_country = pd.DataFrame(country, columns=["Code", "Name", "Continent", "Population", "HeadOfState"])
    return df_country

现在当你调用这个函数时,它不会第二次执行它,但是你会得到你想要的值

def assignment():
    ## do something here
    if Choice == 6:
         # No need to do the name check
        x = SQL_Country()
        if x is None:
            print()

第二个函数缺少参数:

def assignment(df):

您应该在SQLúu County内返回df而不是global:

        global df 
        df = pd.DataFrame(country, columns=["Code", "Name", "Continent", "Population", "HeadOfState"])

应该是:

        return pd.DataFrame(country, columns=["Code", "Name", "Continent", "Population", "HeadOfState"])

然后使用:

df = SQL_Country()

import functools

@functools.lru_cache(maxsize=1)
def SQL_Country():
   ...

这样,数据库获取只完成一次。

In [11]: @functools.lru_cache(maxsize=1)
         def foo():
             print(1)
             return 2

In [12]: foo()
1
Out[12]: 2

In [13]: foo()
Out[13]: 2

我觉得你应该 review python functions

您定义的函数

def SQL_Country():

但是,当您要使用函数时,您要在以下位置提供参数(该参数不应作为函数输入):

df = SQL_Country(df)

此外,您的功能:

def assignment():

可能还应该输入一个数据帧,使其看起来像:

def assignment(df):

此时,对函数的后续调用将是:

assignment(df)

assignment()