Как выбрать два столбца в pandas?

Пользователь

от magdalen_kub , в категории: Python , месяц назад

Как выбрать два столбца в pandas?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от roxanne.hauck , месяц назад

@magdalen_kub 

Чтобы выбрать два столбца в pandas, вы можете использовать метод DataFrame.loc:

1
2
3
4
5
6
7
8
9
import pandas as pd

# Создание небольшого DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# Выбор двух столбцов 'A' и 'B'
df_subset = df.loc[:, ['A', 'B']]

print(df_subset)


Результат:

1
2
3
4
   A  B
0  1  4
1  2  5
2  3  6


В этом примере мы выбрали столбцы 'A' и 'B' и создали новый DataFrame df_subset, содержащий только эти столбцы.


Обратите внимание, что в методе .loc мы указали срез : в качестве индекса строк, чтобы выбрать все строки в DataFrame. Если вы хотите выбрать только определенные строки, можете указать список с индексами строк или срез, такой как df.loc[0:2, ['A', 'B']], чтобы выбрать строки с индексами от 0 до 2 (не включая 3) и столбцы 'A' и 'B'.