Получить индекс максимального значения в столбце DataFrame

Опубликовано: 28 Марта, 2022

Pandas DataFrame - это двумерная потенциально неоднородная структура табличных данных с изменяемым размером и помеченными осями (строки и столбцы).

Посмотрим, как получить индекс максимального значения в столбце DataFrame.

Observe this dataset first. We’ll use ‘Weight’ and ‘Salary’ columns of this data in order to get the index of maximum values from a particular column in Pandas DataFrame.

# importing pandas module 
import pandas as pd 
    
# making data frame 
  
df.head(10)

Код №1: проверьте индекс, при котором присутствует максимальное значение веса.

# importing pandas module 
import pandas as pd 
    
# making data frame 
df = pd.read_csv("nba.csv")
  
# Returns index of maximum weight
df[["Weight"]].idxmax()

Выход:

We can verify whether the maximum value is present in index or not.

   
# importing pandas module 
import pandas as pd 
     
# making data frame 
df = pd.read_csv("nba.csv")
   
# from index 400 to 409
df.iloc[400:410]

Output:

 
Code #2: Let’s insert a new row at index 0, having maximum salary and then varify.

# importing pandas module 
import pandas as pd 
     
# making data frame 
df = pd.read_csv("nba.csv")
  
new_row = pd.DataFrame({"Name":"Geeks", "Team":"Boston", "Number":3,
                        "Position":"PG", "Age":33, "Height":"6-2",
                        "Weight":189, "College":"MIT", "Salary":999999999}
                         , index=[0])
   
df = pd.concat([new_row, df]).reset_index(drop=True)
df.head(5)

Выход:

Now, let’s check if the maximum salary is present at index 0 or not.

# Returns index of minimum salary
df[["Salary"]].idxmax()

Выход:

Внимание компьютерщик! Укрепите свои основы с помощью базового курса программирования Python и изучите основы.

Для начала подготовьтесь к собеседованию. Расширьте свои концепции структур данных с помощью курса Python DS. А чтобы начать свое путешествие по машинному обучению, присоединяйтесь к курсу Машинное обучение - базовый уровень.