Як вибрати кілька стовпців у pandas (з прикладами)
Є три основні методи, які можна використати для вибору кількох стовпців із pandas DataFrame:
Спосіб 1: виберіть стовпці за індексом
df_new = df. iloc [:,[0,1,3]]
Спосіб 2: виберіть стовпці в діапазоні індексів
df_new = df. iloc [:, 0:3]
Спосіб 3: Виберіть стовпці за назвою
df_new = df[[' col1 ', ' col2 ']]
У наведених нижче прикладах показано, як використовувати кожен метод із такими pandas DataFrame:
import pandas as pd
#createDataFrame
df = pd. DataFrame ({' points ': [25, 12, 15, 14, 19, 23, 25, 29],
' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12],
' blocks ': [4, 7, 7, 6, 5, 8, 9, 10]})
#view DataFrame
df
points assists rebounds blocks
0 25 5 11 4
1 12 7 8 7
2 15 7 10 7
3 14 9 6 6
4 19 12 6 5
5 23 9 5 8
6 25 9 9 9
7 29 4 12 10
Спосіб 1: виберіть стовпці за індексом
Наступний код показує, як вибрати стовпці в позиціях індексу 0, 1 і 3:
#select columns in index positions 0, 1, and 3
df_new = df. iloc [:,[0,1,3]]
#view new DataFrame
df_new
points assists blocks
0 25 5 4
1 12 7 7
2 15 7 7
3 14 9 6
4 19 12 5
5 23 9 8
6 25 9 9
7 29 4 10
Зауважте, що вибрано стовпці з індексами 0, 1 і 3.
Примітка : перший стовпець pandas DataFrame розташований у позиції 0.
Спосіб 2: виберіть стовпці в діапазоні індексів
Наступний код показує, як вибрати стовпці в діапазоні індексів від 0 до 3:
#select columns in index range 0 to 3
df_new = df. iloc [:, 0:3]
#view new DataFrame
df_new
points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6
5 23 9 5
6 25 9 9
7 29 4 12
Зауважте, що стовпець, розташований в останньому значенні діапазону (3), не буде включений у вивід.
Спосіб 3: Виберіть стовпці за назвою
Наступний код показує, як вибрати стовпці за назвою:
#select columns called 'points' and 'blocks'
df_new = df[[' points ', ' blocks ']]
#view new DataFrame
df_new
point blocks
0 25 4
1 12 7
2 15 7
3 14 6
4 19 5
5 23 8
6 25 9
7 29 10
Додаткові ресурси
У наступних посібниках пояснюється, як виконувати інші типові операції в pandas:
Як перерахувати всі назви стовпців у Pandas
Як видалити стовпці в Pandas
Як перетворити індекс на стовпець у Pandas