如何獲取連續熊貓中具有最高值的列的名稱

Question

print(df_merged)

          VolumeOIH  VolumeKRE   SPY close  difference  
    0       300.0    51852.0     430.00       -6.67  
    1       700.0    54027.0     437.85        5.63  
    2       500.0    46517.0     435.72       -4.31  
    3      5100.0    43994.0     432.22        0.66  
    4     12400.0    52993.0     419.45      -10.84  
    ..        ...        ...        ...         ...  
    64     8677.0    11411.0     409.62       -5.06  
    65      911.0    17334.0     417.37        8.51  
    66     1990.0    16353.0     410.54       -2.34  
    67     1687.0    13611.0     411.77       -2.54  
    68     2307.0    11291.0     415.70        6.72  

如您所見，我的列有不同的名稱。 我想創建一個新列，它將簡單地顯示具有最高音量的列的名稱。

如果我們取第 0 行，則最高交易量是 VolumeKRE。 我想要創建一個新列，其中第一行的值將顯示：VolumeKRE。

Answer 1

以下命令將為您提供一個具有最大列名稱的列：

df_merged.idxmax(axis=1)

然后你可以將它合並到你的原始數據框