как сделать срез по столбцам pandas

Аватар пользователя Ivan Mamtsev
Ivan Mamtsev
29 мая 2024

Для того чтобы сделать срез по столбцам в pandas, можно воспользоваться методом loc или методом iloc.

  1. С использованием метода loc:
import pandas as pd

# создание DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

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

print(sliced_df)

Результат:

   A   B
0  1   6
1  2   7
2  3   8
3  4   9
4  5  10
  1. С использованием метода iloc:
import pandas as pd

# создание DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

# срез по столбцам с помощью iloc
sliced_df = df.iloc[:, [0, 1]]

print(sliced_df)

Результат:

   A   B
0  1   6
1  2   7
2  3   8
3  4   9
4  5  10
0 0