Table of Contents
Find Column Name with Maximum Value for Each Row
Home Backend Development Python Tutorial How to Find the Column Name with the Maximum Value for Each Row in a DataFrame?

How to Find the Column Name with the Maximum Value for Each Row in a DataFrame?

Nov 14, 2024 pm 04:28 PM

How to Find the Column Name with the Maximum Value for Each Row in a DataFrame?

Find Column Name with Maximum Value for Each Row

Many datasets have rows representing different entities, each with multiple columns representing data about the entities. Sometimes, it becomes necessary to identify the specific column that contains the maximum value for each row. This task can be achieved using the idxmax() function.

In the given example, we have a DataFrame with four columns: "Communications and Search," "Business," "General," and "Lifestyle." We want to create a new column, "Max," that contains the column name corresponding to the maximum value of each row.

To do this, we can use idxmax(axis=1) to find the column index with the maximum value:

df.idxmax(axis=1)

# Output:
0    Communications
1          Business
2    Communications
3    Communications
4          Business
dtype: object
Copy after login

This gives us the column labels, but we can convert them to the corresponding column names:

df['Max'] = df.idxmax(axis=1)
Copy after login

The resulting DataFrame will look like this:

communications and search business general lifestyle max
0.745763 0.050847 0.118644 0.084746 Communications
0.333333 0.000000 0.583333 0.083333 Business
0.617021 0.042553 0.297872 0.042553 Communications
0.435897 0.000000 0.410256 0.153846 Communications
0.358974 0.076923 0.410256 0.153846 Business

Note that idxmax() can also be used to find the row index at which the maximum value occurs in each column, using df.idxmax() (or df.idxmax(axis=0)).

The above is the detailed content of How to Find the Column Name with the Maximum Value for Each Row in a DataFrame?. For more information, please follow other related articles on the PHP Chinese website!

Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn

Hot Article Tags

Notepad++7.3.1

Notepad++7.3.1

Easy-to-use and free code editor

SublimeText3 Chinese version

SublimeText3 Chinese version

Chinese version, very easy to use

Zend Studio 13.0.1

Zend Studio 13.0.1

Powerful PHP integrated development environment

Dreamweaver CS6

Dreamweaver CS6

Visual web development tools

SublimeText3 Mac version

SublimeText3 Mac version

God-level code editing software (SublimeText3)

How Do I Use Beautiful Soup to Parse HTML? How Do I Use Beautiful Soup to Parse HTML? Mar 10, 2025 pm 06:54 PM

How Do I Use Beautiful Soup to Parse HTML?

Image Filtering in Python Image Filtering in Python Mar 03, 2025 am 09:44 AM

Image Filtering in Python

How to Download Files in Python How to Download Files in Python Mar 01, 2025 am 10:03 AM

How to Download Files in Python

How to Use Python to Find the Zipf Distribution of a Text File How to Use Python to Find the Zipf Distribution of a Text File Mar 05, 2025 am 09:58 AM

How to Use Python to Find the Zipf Distribution of a Text File

How to Work With PDF Documents Using Python How to Work With PDF Documents Using Python Mar 02, 2025 am 09:54 AM

How to Work With PDF Documents Using Python

How to Cache Using Redis in Django Applications How to Cache Using Redis in Django Applications Mar 02, 2025 am 10:10 AM

How to Cache Using Redis in Django Applications

How to Perform Deep Learning with TensorFlow or PyTorch? How to Perform Deep Learning with TensorFlow or PyTorch? Mar 10, 2025 pm 06:52 PM

How to Perform Deep Learning with TensorFlow or PyTorch?

How to Implement Your Own Data Structure in Python How to Implement Your Own Data Structure in Python Mar 03, 2025 am 09:28 AM

How to Implement Your Own Data Structure in Python

See all articles