Pandas: 특정 문자열이 포함된 열을 선택하는 방법

다음 방법을 사용하여 Pandas DataFrame에서 특정 문자열이 포함된 열을 선택할 수 있습니다.

방법 1: 특정 문자열이 포함된 열 선택

 df. filter (regex=' string1 ')

방법 2: 여러 문자열 중 하나를 포함하는 열 선택

 df. filter (regex=' string1|string2|string3 ')

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' mavs ': [10, 12, 14, 15, 19, 22, 27],
                   ' cavs ': [18, 22, 19, 14, 14, 11, 20],
                   ' hornets ': [5, 7, 7, 9, 12, 9, 14],
                   ' spurs ': [10, 12, 14, 13, 13, 19, 22],
                   ' net ': [10, 14, 25, 22, 25, 17, 12]})

#view DataFrame
print (df)

   mavs cavs hornets spurs nets
0 10 18 5 10 10
1 12 22 7 12 14
2 14 19 7 14 25
3 15 14 9 13 22
4 19 14 12 13 25
5 22 11 9 19 17
6 27 20 14 22 12

예시 1: 특정 문자열이 포함된 열 선택

다음 코드는 filter() 함수를 사용하여 이름 어딘가에 “avs” 문자열이 포함된 열만 선택하는 방법을 보여줍니다.

 #select columns that contain 'avs' in the name
df2 = df. filter (regex=' avs ')

#view DataFrame
print (df2)

   mavs cavs
0 10 18
1 12 22
2 14 19
3 15 14
4 19 14
5 22 11
6 27 20

이름에 “avs”가 포함된 열만 반환됩니다.

이 경우 “mavs” 및 “cavs”가 반환되는 유일한 열입니다.

예 2: 여러 문자열 중 하나를 포함하는 열 선택

다음 코드는 filter() 함수를 사용하여 이름 어딘가에 “avs” 또는 “ets”가 포함된 열만 선택하는 방법을 보여줍니다.

 #select columns that contain 'avs' in the name
df2 = df. filter (regex=' avs|ets ')

#view DataFrame
print (df2)

   mavs cavs hornets nets
0 10 18 5 10
1 12 22 7 14
2 14 19 7 25
3 15 14 9 22
4 19 14 12 25
5 22 11 9 17
6 27 20 14 12

이름에 “avs” 또는 “ets”가 포함된 열만 반환됩니다.

세로 막대( | )는 Pandas의 ” OR ” 연산자입니다.

다양한 문자열 중 하나가 포함된 열을 선택하려면 이러한 “OR” 연산자를 원하는 만큼 연결하세요.

추가 리소스

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

Pandas: DataFrame 앞으로 열을 이동하는 방법
Pandas:열에 문자열이 포함되어 있는지 확인하는 방법
Pandas: DataFrame에 빈 열을 추가하는 방법(예제 3개)