首页 > 后端开发 > Python教程 > 如何检索 Pandas DataFrame 中每行具有最大值的列的名称?

如何检索 Pandas DataFrame 中每行具有最大值的列的名称?

Mary-Kate Olsen
发布: 2024-11-28 12:41:12
原创
641 人浏览过

How to retrieve the name of the column with the maximum value for each row in a Pandas DataFrame?

检索每行的最大值列名称

在由各种列和行组成的 DataFrame 中,一个常见的任务是识别每行具有最大值的列。考虑以下 DataFrame:

通信和搜索业务一般生活方式<br>0 0.745763 0.050847 0.118644 0.084746<br>0 0.333333 0.000000 0.583333 0.083333<br>0 0.617021 0.042553 0.297872 0.042553<br>0 0.435897 0.000000 0.410256 0.153846<br>0 0.358974 0.076923 0.410256 0.153846<br>
登录后复制

我们的目标是创建一个标记为“Max”的新列,其中包含与每行中的最大值关联的列名称。所需的输出类似于以下内容:

Communications and Search Business General Lifestyle Max<br>0 0.745763 0.050847 0.118644 0.084746 Communications <br>0 0.333333 0.000000 0.583333 0.083333 商业 <br>0 0.617021 0.042553 0.297872 0.042553 通讯 <br>0 0.435897 0.000000 0.410256 0.153846 通讯 <br>0 0.358974 0.076923 0.410256 0.153846 商业 <br>
登录后复制

要实现此目的,我们可以使用 idxmax 函数:

import pandas as pd

# Create a DataFrame
df = pd.DataFrame({
    'Communications and Search': [0.745763, 0.333333, 0.617021, 0.435897, 0.358974],
    'Business': [0.050847, 0.000000, 0.042553, 0.000000, 0.076923],
    'General': [0.118644, 0.583333, 0.297872, 0.410256, 0.410256],
    'Lifestyle': [0.084746, 0.083333, 0.042553, 0.153846, 0.153846]
})

# Find the column index with the maximum value in each row
max_column_idxs = df.idxmax(axis=1)

# Create a new column with the column names
df['Max'] = max_column_idxs

# Display the updated DataFrame
print(df)
登录后复制

通过使用 idxmax 函数并将 axis 参数设置为 1 ,我们确定每行具有最大值的列索引。然后使用此信息创建一个名为“Max”的新列,该列标识每行最大值的相应列名称。生成的 DataFrame 显示了请求的格式。

以上是如何检索 Pandas DataFrame 中每行具有最大值的列的名称?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板