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

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

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

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

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

от 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'.

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

от krista , 7 месяцев назад

@magdalen_kub 

Также можно использовать метод DataFrame[['столбец1', 'столбец2']]:


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[['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. Этот способ также позволяет выбирать столбцы по их имени.