как разделить столбцы pandas

Аватар пользователя Elena Gromova
Elena Gromova
10 июля 2024

Для разделения столбцов в объекте DataFrame библиотеки pandas можно использовать методы .iloc[] или .loc[].

Например, для разделения столбцов с индексами от 0 до 2 можно использовать следующий код:

import pandas as pd

data = {'A': [1, 2, 3],
        'B': [4, 5, 6],
        'C': [7, 8, 9]}

df = pd.DataFrame(data)

# разделение столбцов с помощью метода iloc[]
df_subset = df.iloc[:, 0:2]

# разделение столбцов с помощью метода loc[]
df_subset = df.loc[:, 'A':'B']

Таким образом, в переменной df_subset будет содержаться DataFrame с двумя первыми столбцами из исходного DataFrame df.

Также можно разделить столбцы по названиям с помощью следующего кода:

# разделение столбцов по названиям
df_subset = df[['A', 'C']]

Этот код создаст новый DataFrame df_subset, содержащий только столбцы 'A' и 'C' из исходного DataFrame df.

Таким образом, существует несколько способов разделения столбцов в объекте DataFrame библиотеки pandas, в зависимости от ваших потребностей и предпочтений.

0 0
Познакомьтесь с основами аналитики данных бесплатно