как сделать срез по столбцам pandas
Ответы
Ivan Mamtsev
29 мая 2024
Для того чтобы сделать срез по столбцам в pandas, можно воспользоваться методом loc
или методом iloc
.
- С использованием метода
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
- С использованием метода
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