>  기사  >  백엔드 개발  >  Python으로 CSV 파일을 처리할 때 헤더를 제외하는 방법은 무엇입니까?

Python으로 CSV 파일을 처리할 때 헤더를 제외하는 방법은 무엇입니까?

Barbara Streisand
Barbara Streisand원래의
2024-10-31 18:28:29525검색

How to Exclude Headers When Processing CSV Files with Python?

Python으로 CSV 파일을 처리할 때 헤더 제외

CSV 파일을 처리하는 동안 특정 작업에서 헤더를 제외해야 할 수도 있습니다. 이는 Python의 CSV 판독기 및 기록기 모듈을 활용하여 달성할 수 있습니다.

제공된 코드에서는 헤더 행을 건너뛰는 것이 좋습니다. 행 변수를 1로 초기화하는 대신, 더 간단한 접근 방식은 나머지를 처리하기 전에 첫 번째 행을 건너뛰는 것입니다. 이는 다음과 같이 수행할 수 있습니다.

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    next(reader, None)  # Skip the headers
    writer = csv.writer(outfile)
    for row in reader:
        # Process each row
        writer.writerow(row)</code>

next(reader, None)를 호출하면 CSV 파일의 첫 번째 행이 검색되고 삭제됩니다. 그러면 다음 행을 처리하여 출력 파일에 쓸 수 있습니다.

또한 컨텍스트 관리자를 사용하여 파일 열기 및 닫기를 자동으로 처리함으로써 코드를 단순화할 수 있습니다.

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)
    # Skip the headers
    if headers := next(reader, None):
        writer.writerow(headers)</code>

여기서 코드에서 선택적 headers 변수는 건너뛴 헤더를 수신하여 사용자가 원하는 경우 처리되지 않은 헤더를 출력 파일에 쓸 수 있도록 합니다.

위 내용은 Python으로 CSV 파일을 처리할 때 헤더를 제외하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.