Найдите сумму и максимальное значение двух столбцов в файле Excel с помощью Pandas

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

В этих статьях мы обсудим, как читать данные из Excel, выполнять некоторые математические операции и сохранять их в новом столбце в DataFrame. Предположим, наш файл Excel выглядит так.

sample_data.xlsx

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

Подход :

  • Модуль импорта Pandas.
  • Считайте данные из Excel.
  • Создайте новый столбец для хранения суммы и максимума.
  • Установите индекс каждого столбца для доступа к элементу.
  • Сохраните сумму двух столбцов в новом столбце.
  • И сохраните максимальное число из двух столбцов в столбце.
  • Отобразить DataFrame.

Step 1: Importing module and reading from excel.

Python3

# import module
import pandas as pd
  
# read from excel
# and store in a DataFram
df = pd.read_excel("excel_work/book_sample.xlsx")
df

Выход :

Step 2: Create a new column for storing sum and max

Python3

# creation new column
df["Total"] = None
df["Maximum"] = None
df

Выход :

Step 3: Set an index for accessing the required column.

Python3



# Set index for each column
index_selling=df.columns.get_loc("Selling Price")
index_cost=df.columns.get_loc("Cost price")
index_total=df.columns.get_loc("Total")
index_max=df.columns.get_loc("Maximum")
  
print(index_selling,index_cost,index_total,index_max)

Выход :

 2 3 4 5

Step 4: Select each row and add a column and find maximum

Python3

for row in range(0, len(df)):
    df.iat[row, index_total] = df.iat[row,
                                      index_selling] + df.iat[row, index_cost]
      
    if df.iat[row, index_selling] > df.iat[row, index_cost]:
        df.iat[row, index_max] = df.iat[row, index_selling]
    else:
        df.iat[row, index_max] = df.iat[row, index_cost]
df

Выход :

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

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