>  기사  >  백엔드 개발  >  데이터프레임에서 지정된 문자열을 포함하는 열 이름을 추출하는 방법은 무엇입니까?

데이터프레임에서 지정된 문자열을 포함하는 열 이름을 추출하는 방법은 무엇입니까?

Patricia Arquette
Patricia Arquette원래의
2024-10-20 14:02:29416검색

How to Extract Column Names Containing a Specified String in Dataframes?

지정된 문자열이 포함된 열 이름 추출

데이터 프레임에서는 특정 이름으로 열에 액세스하는 것이 중요할 수 있습니다. 이 질문은 정확히 일치하지 않더라도 이름에 특정 문자열이 포함된 열을 식별해야 하는 시나리오를 다룹니다. 제공된 예는 'spike-2', 'hey spiked' 및 'spiked-in'과 같은 열 이름에서 'spike'를 검색하는 것입니다.

해결책:

이를 달성하려면 DataFrame의 열을 반복하는 간단한 접근 방식이 필요합니다.

<code class="python">for col in df.columns:
    if 'spike' in col:
        # Do something with the column name</code>

이 솔루션에서는 각 열 이름을 검사하여 대상 문자열이 포함되어 있는지 확인합니다. 일치하는 항목이 발견되면 추가 사용을 위해 열 이름을 변수에 저장할 수 있습니다.

또 다른 옵션은 목록 이해 및 필터링을 활용하여 일치하는 열만 포함하는 새 데이터 프레임을 만드는 것입니다.

<code class="python">spike_cols = [col for col in df.columns if 'spike' in col]
df2 = df.filter(regex='spike')</code>

첫 번째 줄은 'spike'가 포함된 열 이름 목록을 생성하고, 두 번째 줄은 해당 열만 포함하도록 데이터프레임을 필터링합니다.

이러한 기술을 활용하면 열을 효율적으로 식별하고 액세스할 수 있습니다. 이름에 특정 문자열이 포함되어 분석 능력이 넓어집니다.

위 내용은 데이터프레임에서 지정된 문자열을 포함하는 열 이름을 추출하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.