在执行merg时重塑数据帧

2024-04-30 02:13:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个pandas.DataFramedf和一些元数据,其中有一个IDColumnValue,我想与另一个df结合使用,例如:

df_map = pd.DataFrame({"ID" : [3, 7, 17], "Column" : ["A1", "B7", "C17"], 
                       "Value" : ["ValA1", "ValB7", "ValC17"]})

我想把上面的与下面的df结合起来,其中列名称与上面Column中的行条目匹配,下面df中的行值与上面ID行值匹配。你知道吗

df_main = pd.DataFrame({"A1" : [3, 6], "A5" : [5, 10], "B7" : [7, 14] , 
                        "C17" : [17, 34], "C19" : [19, 38] })

因此,我希望这些合并到df's中,这样我就可以基于Value列对其进行重塑,将它们添加为ID's匹配的附加维度,即df_result = combine(df_map, df_main)

我基本上预期结果如下

df_result = pd.DataFrame({"A1" : [3, 6], "A5" : [5, 10], "B7" : [7, 14] ,
                          "C17" : [17, 34], "C19" : [19, 38], "Value A1" : ["ValA1", None],
                         "Value B7" : ["ValB7", None], "Value C17" : ["ValC17", None ]})

Out[30]:
   A1  A5  B7  C17  C19 Value A1 Value B7 Value C17
0   3   5   7   17   19    ValA1    ValB7    ValC17
1   6  10  14   34   38     None     None      None

不确定在pandas中最好的方法是什么?


Tags: noneiddataframedfvaluea1columnpd
2条回答

具有Series.mappandas.concat的替代解决方案:

df2=pd.concat([df_main.T[key].map(df_map.set_index('ID')['Value']) for key in df_main.index.tolist()],axis=1).T.add_prefix('Value_')
df_main=pd.concat([df_main,df2],axis=1)
df_main.dropna(how='all',axis=1,inplace=True)
print(df_main)

   A3  A5  B7  C17  C19 Value_A3 Value_B7 Value_C17
0   3   5   7   17   19    ValA1    ValB7    ValC17
1   6  10  14   34   38      NaN      NaN       NaN

首先^{}index转换为列,以避免在^{}中丢失,然后使用^{}^{}重新整形,使用^{}仅删除缺少的列,最后使用mapMultiIndex展平:

df = (df_main.reset_index()
             .melt('index',var_name='Column', value_name='ID')
             .merge(df_map, how='left')
             .set_index(['index', 'Column'])
             .unstack()
             .rename_axis(None)
             .dropna(how='all', axis=1))
df.columns = df.columns.map('_'.join)
print (df)
   ID_A1  ID_A5  ID_B7  ID_C17  ID_C19 Value_A1 Value_B7 Value_C17
0      3      5      7      17      19    ValA1    ValB7    ValC17
1      6     10     14      34      38      NaN      NaN       NaN

相关问题 更多 >