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 중국어 웹사이트의 기타 관련 기사를 참조하세요!