본문 바로가기
Computer Science/[20-3,4] Python Basic

[Python] csv 파일에서 특정 단어를 가진 열 찾아보기

by gojw 2020. 4. 22.
import csv

def csv_word(filename, word):
    f = open(filename, 'r')
    content = csv.reader(f)

    found = []
    for line in content:
        if isinstance(word, str):
            if word in line:
                found.append(line)
        else:
            if len(set(word) & set(line)) == len(word):
                found.append(line)

    final = ''
    for item in found:
        final += str(item) + '\n'

    return final if len(found) != 0 else f"'{word}' doesn't exist"

→ 두번째 인자로 str을 받으면 그 단어를 포함한 열들을 리턴하고,

컨테이너 자료형을 받으면 안의 요소들을 모두 포함한 열들을 리턴한다.

댓글