language/python

파이썬에서 엑셀 사용하기(조건에 맞는 값 추출하기) #2

superminy 2023. 9. 12. 16:57
728x90

파이썬을 이용해 엑셀에서 특정 조건에 맞는 데이터를 추출하는 방법을 알아보겠습니다.

 

아래 코드는 openpyxl 라이브러리를 사용하여 엑셀 파일에서

 

특정 조건에 맞는 데이터를 추출할 수 있습니다.

 

from openpyxl import load_workbook

# 엑셀 파일 로드
workbook = load_workbook(filename='test_excel.xlsx')

# 원하는 시트 선택
sheet = workbook['Sheet1']

# 추출한 데이터를 저장할 리스트 생성
filtered_data = []

# 데이터 추출을 위한 반복문 실행
for row in sheet.iter_rows():  # 첫 번째 행은 헤더로 가정하고 건너뜁니다.
    # 조건에 맞는지 확인하고 해당하는 경우 filtered_data 리스트에 추가합니다.
    if row[0].value == '장미과':  # 예시로 첫 번째 열의 값이 '조건값'인 경우만 추출합니다.
        data_row = [cell.value for cell in row]  # 해당 행의 모든 셀 값을 가져와서 리스트로 만듭니다.
        filtered_data.append(data_row)

# 결과 출력
for data in filtered_data:
    print(data)

# 파일 닫기
workbook.close()

과일 종류가 있는 엑셀데이터
조건에 있는 장미과 데이터를 추출

위 코드에서는 iter_rows() 함수를 사용하여 각 행을 순회하고

 

조건에 맞으면 해당 행의 값을 가져와서 filtered_data 리스트에 추가합니다.

 

이후 filtered_data 리스트를 원하는 방식으로 활용할 수 있습니다.

 

추가적인 필터링이나 다양한 조작을 위해서는

 

파이썬의 제어문과 연산자 등을 활용하여 코드를 수정할 수 있습니다.

 

728x90
반응형