将多个数据帧保存到 Python 的环境中

Save multiple dataframes into the environment in Python

我和 有类似的问题,但最初的问题是制作多个 csv 输出。就我而言,我想知道是否有办法通过循环将多个数据帧输出到环境中,以便我可以进行一些数据分析。

us = df[df['country_code'].str.match("US")]
mx = df[df['country_code'].str.match("MX")]
ca = df[df['country_code'].str.match("CA")]
au = df[df['country_code'].str.match("AU")]

您可以使用与发布的 link 相同的代码,但将不同的 dfs 保存到字典中:

codes = ['US', 'MX', 'CA', 'AU']
result_dict = {}
for code in codes:
    temp = df.query(f'country_code.str.match("{code}")')
    result_dict[code] = temp

您可以创建 for 并像下面这样检查并创建 dict 以匹配:

df = pd.DataFrame({'country_code': ['US','MX', 'CA', 'AU']})
codes = ['US', 'MX', 'CA', 'AU']
out = {code : df[df['country_code'].str.match(code)] for code in codes}

输出:

>>> out["US"]
     country_code
0    US

>>> type(out["US"])
pandas.core.frame.DataFrame


>>> out["CA"]
     country_code
2    CA