蟒蛇 3 |动态创建数据框后如何按名称打印数据框？

Question

我有一个 Dataframe df_main 包含以下列：

ID	Category	Time	Status	XYZ
1	A	value	value	value
2	B	value	value	value
3	C	value	value	value
4	D	value	value	value
5	E	value	value	value

使用以下代码，我根据表中的类别创建了新的数据框。我已经创建了一个 Dataframe 字典并以这种格式创建了数据框 df_A、df_B、df_C... 我已将该行存储在与类别名称等效的新数据框中。因此，df_A 将拥有来自df_main 的行，该行具有类别值“A”。

代码：

dict_of_df = {} # initialize empty dictionary

i=0
for index, row in df_main.iterrows():
    
    if i<5:
        newname = df_main['Category'].values[i]
        dict_of_df["df_{}".format(newname)] = row
        
    i=i+1

我想通过数据框名称打印数据框，而不是通过迭代字典。 应该是这样的：

print(df_A)
print(df_B)
print(df_C)
print(df_D)
print(df_E)

我怎样才能做到这一点？ 不使用字典的解决方案也可以。只要我能够将特定类别的一行存储在特定于类别名称的新数据框中并使用数据框名称打印它，任何解决方案都可以。

如果需要更多详细信息，请告诉我。

编辑：

这个链接有点类似于我的用例： Using String Variable as variable name

我想具体到数据帧，因为我的最终目标是按名称打印数据帧。

该链接的答案中提到的方法特定于变量，并且需要使用 exec 数据帧方法的不同代码解决方案。

此代码背后的想法是将其包含在 Power BI 中。在 Power BI 中使用 python 脚本获取源接受数据帧作为表，为此，我必须在代码中声明或打印数据帧。

Answer 1

更改您的数据框源

import pandas as pd

df_main = pd.read_excel('main.xlsx') # use data source as per your requirement 

dict_of_df = {}  # initialize empty dictionary

i = 0
for index, row in df_main.iterrows():
    if i < 5:
        print('df_'+row['Category'])
        newname = df_main['Category'].values[i]
        dict_of_df["df_{}".format(newname)] = row

    i = i + 1