프로그래밍/python
dataframe에서 값이 일치하는 열이 있는 행을 검색하는 코드
이휘재123
2023. 5. 10. 17:24
반응형
import pandas as pd
# 샘플 데이터프레임 생성
df = pd.DataFrame({'A': [1, 2, 3, 4], 'B': [5, 6, 7, 8], 'C': [9, 10, 11, 12]})
# 값이 일치하는 열이 있는 행 검색
search_value = 2 # 검색할 값
search_column = 'A' # 검색할 열
result = df.loc[df[search_column] == search_value]
print(result)
위와 같이 검색을 한다면 A열이 2인 행, 즉 1행을 출력하게 된다.
반응형