Панды: как найти разницу между двумя строками
Вы можете использовать функцию DataFrame.diff() , чтобы найти разницу между двумя строками в DataFrame pandas.
Эта функция использует следующий синтаксис:
DataFrame.diff(периоды=1, ось=0)
Золото:
- периоды: количество предыдущих строк для расчета разницы.
- ось: найдите разницу в строках (0) или столбцах (1).
Следующие примеры показывают, как использовать эту функцию на практике.
Пример 1. Найдите разницу между каждой предыдущей строкой.
Предположим, у нас есть следующий DataFrame pandas:
import pandas as pd #createDataFrame df = pd. DataFrame ({' period ': [1, 2, 3, 4, 5, 6, 7, 8], ' sales ': [12, 14, 15, 15, 18, 20, 19, 24], ' returns ': [2, 2, 3, 3, 5, 4, 4, 6]}) #view DataFrame df period sales returns 0 1 12 2 1 2 14 2 2 3 15 3 3 4 15 3 4 5 18 5 5 6 20 4 6 7 19 4 7 8 24 6
Следующий код показывает, как найти разницу между каждой текущей строкой DataFrame и предыдущей строкой:
#add new column to represent sales differences between each row df[' sales_diff '] = df[' sales ']. diff () #view DataFrame df period sales returns sales_diff 0 1 12 2 NaN 1 2 14 2 2.0 2 3 15 3 1.0 3 4 15 3 0.0 4 5 18 5 3.0 5 6 20 4 2.0 6 7 19 4 -1.0 7 8 24 6 5.0
Обратите внимание, что мы также можем найти разницу между несколькими предыдущими строками. Например, следующий код показывает, как найти разницу между каждой текущей строкой и строкой, появившейся тремя строками ранее:
#add new column to represent sales differences between current row and 3 rows earlier df[' sales_diff '] = df[' sales ']. diff (periods= 3 ) #view DataFrame df period sales returns sales_diff 0 1 12 2 NaN 1 2 14 2 NaN 2 3 15 3 NaN 3 4 15 3 3.0 4 5 18 5 4.0 5 6 20 4 5.0 6 7 19 4 4.0 7 8 24 6 6.0
Пример 2. Найдите разницу по условию
Мы также можем фильтровать DataFrame для отображения строк, в которых разница между текущей строкой и предыдущей строкой меньше или больше определенного значения.
Например, следующий код возвращает только те строки, в которых значение текущей строки меньше значения предыдущей строки:
import pandas as pd #createDataFrame df = pd. DataFrame ({' period ': [1, 2, 3, 4, 5, 6, 7, 8], ' sales ': [12, 14, 15, 13, 18, 20, 19, 24], ' returns ': [2, 2, 3, 3, 5, 4, 4, 6]}) #find difference between each current row and the previous row df[' sales_diff '] = df[' sales ']. diff () #filter for rows where difference is less than zero df = df[df[' sales_diff ']< 0 ] #view DataFrame df period sales returns sales_diff 3 4 13 3 -2.0 6 7 19 4 -1.0
Дополнительные ресурсы
Как найти уникальные значения в нескольких столбцах в Pandas
Как фильтровать DataFrame Pandas по значениям столбца
Как выбрать строки по индексу в DataFrame Pandas