Панды: как найти разницу между двумя столбцами

Чтобы найти разницу между двумя столбцами в DataFrame pandas, вы можете использовать следующий синтаксис:

 df[' difference '] = df[' column1 '] - df[' column2 ']

Следующие примеры показывают, как использовать этот синтаксис на практике.

Пример 1. Найдите разницу между двумя столбцами

Предположим, у нас есть следующий DataFrame pandas, который показывает общий объем продаж для двух регионов (A и B) за восемь последовательных периодов продаж:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' period ': [1, 2, 3, 4, 5, 6, 7, 8],
                   ' A_sales ': [12, 14, 15, 13, 18, 20, 19, 24],
                   ' B_sales ': [14, 19, 20, 22, 24, 20, 17, 23]})

#view DataFrame
df

period A_sales B_sales
0 1 12 14
1 2 14 19
2 3 15 20
3 4 13 22
4 5 18 24
5 6 20 20
6 7 19 17
7 8 24 23

Следующий код показывает, как рассчитать разницу между продажами в регионе Б и регионе А для каждого периода продаж:

 #add new column to represent difference between B sales and A sales
df[' diff '] = df[' B_sales '] - df[' A_sales ']

#view DataFrame
df

        period A_sales B_sales diff
0 1 12 14 2
1 2 14 19 5
2 3 15 20 5
3 4 13 22 9
4 5 18 24 6
5 6 20 20 0
6 7 19 17 -2
7 8 24 23 -1

Мы также могли бы рассчитать абсолютную разницу в продажах, используя функцию pandas.Series.abs() :

 #add new column to represent absolute difference between B sales and A sales
df[' diff '] = pd. Series . abs (df[' B_sales '] - df[' A_sales '])

#view DataFrame
df

	period A_sales B_sales diff
0 1 12 14 2
1 2 14 19 5
2 3 15 20 5
3 4 13 22 9
4 5 18 24 6
5 6 20 20 0
6 7 19 17 2
7 8 24 23 1

Пример 2. Найдите разницу между столбцами по условию.

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

Например, следующий код возвращает только те строки, в которых продажи в регионе A превышают продажи в регионе B:

 #add new column to represent difference between B sales and A sales
df[' diff '] = df[' B_sales '] - df[' A_sales ']

#display rows where sales in region A is greater than sales in region B
df[df[' diff ']< 0 ]


        period A_sales B_sales diff
6 7 19 17 -2
7 8 24 23 -1

Дополнительные ресурсы

Панды: как найти разницу между двумя строками
Pandas: как группировать и агрегировать по нескольким столбцам