Question

我有两个数据帧df1和df2。 df1就像是具有以下值的字典

Ticker          INDUSTRY_SECTOR
TLRA_Equity     Communications
KAMN_Equity     Industrial
B_Equity        Industrial
ARNC_Equity     Industrial
RC_Equity       Consumer, Non-cyclical
DAR_Equity      Consumer, Non-cyclical

df2具有以下值：

  Date   TLRA_Equity  KAMN_Equity  B_Equity  ARNC_Equity RC_Equity DAR_Equity
1/1/2000  10              20        30          40        50          60
2/1/2000  15              25        35          45        55          65
3/1/2000  17              27        37          47        57          67

我想基于df1数据帧中的INDUSTRY_SECTOR将df2分为3个新数据帧。

日期TLRA_Equity列应位于Communications数据框中
日期，KAMN_Equity，B_Equity，ARNC_Equity列应位于Industrial数据框中
日期，RC_Equity和DAR_Equity列应位于Consumer, Non-cyclical数据框中

预期输出：

Communications数据框

Date   TLRA_Equity  
1/1/2000  10            
2/1/2000  15            
3/1/2000  17

Industrial数据框

Date    KAMN_Equity  B_Equity  ARNC_Equity 
1/1/2000  20          30          40       
2/1/2000  25          35          45       
3/1/2000  27          37          47

Consumer, Non-cyclical数据帧

Date          RC_Equity DAR_Equity
1/1/2000        50          60
2/1/2000        55          65
3/1/2000        57          67

请让我知道如何高效地进行操作。我正在尝试将例如Communications_TLRA_Equity的列名进行连接，然后根据列名的前半部分拆分数据帧。

代码：

col_names = df2.columns.values.tolist()
d_cols = df2.columns.map(df1.set_index('Ticker')['INDUSTRY_SECTOR'].get)
print(d_cols)
df.columns = [d_cols + "_"  str(col) for col in df.columns]

for sector, df_sector in df.columns.str.split('_').str[0].tolist():
     print(sector)
     print(df_sector)

但这很复杂。需要更好的解决方案。

Answer 1

您可以使用groupby创建数据帧列表，

dfs = [df2.set_index('Date')[ticker].reset_index() for ticker in df1.groupby('INDUSTRY_SECTOR').Ticker.apply(list)]


dfs[0]

    Date    TLRA_Equity
0   1/1/2000    10
1   2/1/2000    15
2   3/1/2000    17


dfs[1]

    Date    RC_Equity   DAR_Equity
0   1/1/2000    50      60
1   2/1/2000    55      65
2   3/1/2000    57      67

dfs[2]


    Date    KAMN_Equity B_Equity    ARNC_Equity
0   1/1/2000    20      30          40
1   2/1/2000    25      35          45
2   3/1/2000    27      37          47

Answer 2

请考虑创建一个dict帧，密钥为您的“ INDUSTRY_SECTOR”名称。

首先创建一个Industry_SECTOR字典：股票行情栏

g = df1.groupby('INDUSTRY_SECTOR')['Ticker'].apply(list).to_dict()
print(g)

{'Communications': ['TLRA_Equity'], 'Consumer, Non-cyclical': ['RC_Equity', 'DAR_Equity'], 'Industrial': ['KAMN_Equity', 'B_Equity', 'ARNC_Equity']}

然后创建字典理解：

frames = {k:df2.set_index('Date')[v] for k,v in g.items()}

print(frames)

{'Communications':           TLRA_Equity
Date                 
1/1/2000           10
2/1/2000           15
3/1/2000           17,
 'Consumer, Non-cyclical':           RC_Equity  DAR_Equity
Date                           
1/1/2000         50          60
2/1/2000         55          65
3/1/2000         57          67,
 'Industrial':           KAMN_Equity  B_Equity  ARNC_Equity
Date                                        
1/1/2000           20        30           40
2/1/2000           25        35           45
3/1/2000           27        37           47}

N.B.

正如@Parfait指出的那样，要包含Date列，您需要将其设置为dict理解中的索引。如果您不希望Date作为索引，请使用：

frames = {k:df2.set_index('Date')[v].reset_index() for k,v in g.items()}

根据来自另一个数据框的值将数据框拆分为多个数据框

2 个答案: