열 값을 기반으로 Pandas DataFrames에서 특정 행을 선택하는 방법은 무엇입니까?

Pandas DataFrames의 열 값을 기준으로 행 선택

Pandas DataFrames로 작업할 때 특정 열의 특정 값을 기준으로 행을 필터링해야 하는 경우가 종종 발생합니다. 이는 WHERE column_name = some_value와 같은 필터를 사용하여 행을 검색하는 SQL 쿼리를 모방합니다.

스칼라 값

열 값이 스칼라 값 some_value와 일치하는 행을 선택하려면 같음 연산자 ==를 사용합니다. :

df.loc[df['column_name'] == some_value]

반복 가능한 값

행 선택 열 값이 some_values ​​배열에 있는 경우 isin 메서드를 사용하세요.

df.loc[df['column_name'].isin(some_values)]

조건 결합

논리 & 연산자를 사용하여 여러 조건을 결합할 수 있습니다.

df.loc[(df['column_name'] >= A) & (df['column_name'] <= B)]

참고: 연산자 우선 순위를 보장하려면 괄호를 사용하세요. 맞습니다.

부정

열 값이 some_value와 같지 않은 행을 선택하려면 불평등 연산자 !=:

df.loc[df['column_name'] != some_value]

isin의 경우 다음을 사용하여 결과를 부정합니다. ~:

df = df.loc[~df['column_name'].isin(some_values)]

예

고려하세요 다음 DataFrame:

import pandas as pd

df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
                   'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
                   'C': np.arange(8),
                   'D': np.arange(8) * 2})
print(df)

A가 foo인 행 선택:

print(df.loc[df['A'] == 'foo'])

B가 1 또는 3인 행 선택:

print(df.loc[df['B'].isin(['one', 'three'])])

색인을 생성하고 다음을 사용하여 행을 선택합니다. it:

df = df.set_index(['B'])
print(df.loc['one'])

색인 값이 여러 개인 행 선택:

print(df.loc[df.index.isin(['one', 'two'])])

위 내용은 열 값을 기반으로 Pandas DataFrames에서 특정 행을 선택하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!