@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'.

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