Pandas: loc를 사용하여 여러 열을 선택하는 방법

Pandas의 loc 함수를 사용하여 DataFrame에서 레이블별로 여러 열을 선택할 수 있습니다.

이를 수행하는 가장 일반적인 방법은 다음과 같습니다.

방법 1: 이름으로 여러 열 선택

 df. loc [:,[' col2 ',' col4 ']]

방법 2: 범위의 모든 열 선택

 df. loc [:, ' col2 ':' col4 ']

다음 예에서는 다음 pandas DataFrame에서 실제로 각 메서드를 사용하는 방법을 보여줍니다.

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' points ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' assists ': [11, 8, 10, 6, 6, 5, 9, 12],
                   ' rebounds ': [6, 7, 7, 6, 10, 12, 10, 9]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 5 11 6
1 To 7 8 7
2 To 7 10 7
3 to 9 6 6
4 B 12 6 10
5 B 9 5 12
6 B 9 9 10
7 B 4 12 9

예 1: 이름으로 여러 열 선택

다음 코드는 loc 함수를 사용하여 DataFrame의 “points” 및 “bounces” 열을 선택하는 방법을 보여줍니다.

 #select points and rebounds columns
df. loc [:,[' points ',' rebounds ']]

        rebound points
0 5 6
1 7 7
2 7 7
3 9 6
4 12 10
5 9 12
6 9 10
7 4 9

‘포인트’ 및 ‘리바운드’ 열의 각 행이 반환됩니다.

또한 loc 함수에서 열을 지정하는 순서는 해당 열이 반환되는 순서입니다.

예를 들어, “리바운드” 열을 먼저 반환한 다음 “포인트” 열을 반환할 수 있습니다.

 #select rebounds and points columns
df. loc [:, [' rebounds ', ' points ']]

	rebound points
0 6 5
1 7 7
2 7 7
3 6 9
4 10 12
5 12 9
6 10 9
7 9 4

예 2: 범위의 모든 열 선택

다음 코드는 loc 함수를 사용하여 DataFrame의 “points” 열과 “bounces” 열 사이의 모든 열을 선택하는 방법을 보여줍니다.

 #select all columns between points and rebounds columns
df. loc [:, ' points ': ' rebounds ']

	points assists rebounds
0 5 11 6
1 7 8 7
2 7 10 7
3 9 6 6
4 12 6 10
5 9 5 12
6 9 9 10
7 4 12 9

DataFrame의 “points” 열과 “bounces” 열 사이의 모든 열이 반환됩니다.

참고 : 인덱스 위치로 열을 선택하려면 대신iloc 함수를 사용하세요.

추가 리소스

다음 튜토리얼에서는 Pandas에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

Pandas Loc을 사용하여 여러 조건에 따라 행을 선택하는 방법
Pandas에서 열 값을 기준으로 행을 선택하는 방법
Pandas에서 인덱스로 행을 선택하는 방법