본문 바로가기

카테고리 없음

[Chat GPT] CSV 파일에서 특정 문구 포함된 행을 다른 파일로 저장

Chat GPT 에 Csv 파일에서 특정 문구가 포함된 행을 찾아서 다시 출력하는 파일을 물어봄

Q1. 파이썬으로 csv 파일 열어서 특정 문구가 포함된 열을 찾아내는 스크립트 만들어줘

Q2. 위에서 찾아낸 행을 별도의 CSV 파일로 저장해줘

Q3. 위에서 여러개의 문구를 찾아서 하고 싶다면?

3번 질문에 따른 결과 스크립트임 , 아직 해보지는 못함

 

import csv

# 찾으려는 문구를 리스트 형태로 저장합니다.
search_phrases = ["찾으려는 문구1", "찾으려는 문구2", "찾으려는 문구3"]

# csv 파일을 읽어와서 검색합니다.
with open('파일명.csv', newline='', encoding='utf-8') as csvfile:
    reader = csv.reader(csvfile)
    
    # csv 파일에서 헤더 행을 가져옵니다.
    header = next(reader)
    
    # 검색 결과를 저장할 리스트를 초기화합니다.
    search_results = []
    
    # csv 파일을 한 행씩 읽어와서 검색합니다.
    for row in reader:
        
        # 각 행에서 검색 문구가 있는지 확인합니다.
        for phrase in search_phrases:
            if phrase in row[0]:  # 여기서는 첫번째 열에서 검색합니다.
                search_results.append(row)
                break  # 검색 문구를 찾았으면 다음 행으로 넘어갑니다.
                
# 검색 결과를 새로운 csv 파일로 저장합니다.
with open('검색결과.csv', 'w', newline='', encoding='utf-8') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(header)
    writer.writerows(search_results)