Pandas: як створити стовпець, якщо він не існує

Ви можете використовувати наступний базовий синтаксис, щоб створити стовпець у pandas DataFrame, якщо він ще не існує:

 df[' my_column '] = df. get (' my_column ', df[' col1 '] * df[' col2 ']) 

Цей конкретний синтаксис створює новий стовпець під назвою my_column , якщо він ще не існує в DataFrame і визначається як добуток існуючих стовпців col1 і col2 .

У наступному прикладі показано, як використовувати цей синтаксис на практиці.

Приклад: створіть стовпець у Pandas, якщо він не існує

Припустімо, що у нас є наступні pandas DataFrame:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' day ': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
                   ' sales ': [4, 6, 5, 8, 14, 13, 13, 12, 9, 8, 19, 14],
                   ' price ': [1, 2, 2, 1, 2, 4, 4, 3, 3, 2, 2, 3]})

#view DataFrame
print (df)

    day sales price
0 1 4 1
1 2 6 2
2 3 5 2
3 4 8 1
4 5 14 2
5 6 13 4
6 7 13 4
7 8 12 3
8 9 9 3
9 10 8 2
10 11 19 2
11 12 14 3

Тепер припустімо, що ми спробуємо додати стовпець під назвою price , якщо він ще не існує, і визначимо його як стовпець, де кожне значення дорівнює 100:

 #attempt to add column called 'price'
df[' price '] = df. get (' price ', 100)    

#view updated DataFrame
print (df)

    day sales price
0 1 4 1
1 2 6 2
2 3 5 2
3 4 8 1
4 5 14 2
5 6 13 4
6 7 13 4
7 8 12 3
8 9 9 3
9 10 8 2
10 11 19 2
11 12 14 3

Оскільки стовпець під назвою price вже існує, pandas просто не додає його до DataFrame.

Однак, скажімо, ми спробуємо додати новий стовпець під назвою дохід , якщо він ще не існує, і визначимо його як стовпець, де значення є добутком стовпців продажів і цін:

 #attempt to add column called 'revenue'
df[' revenue '] = df. get (' revenue ', df[' sales '] * df[' price '])

#view updated DataFrame
print (df)

    day sales price revenue
0 1 4 1 4
1 2 6 2 12
2 3 5 2 10
3 4 8 1 8
4 5 14 2 28
5 6 13 4 52
6 7 13 4 52
7 8 12 3 36
8 9 9 3 27
9 10 8 2 16
10 11 19 2 38
11 12 14 3 42

Цей стовпець доходу додано до DataFrame, оскільки він ще не існував.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в pandas:

Як видалити рядки в Pandas DataFrame на основі умови
Як відфільтрувати Pandas DataFrame за кількома умовами
Як використовувати фільтр «NOT IN» у Pandas DataFrame