Python을 사용하여 CSV 파일에서 특정 열을 추출하는 방법은 무엇입니까?

Python csv 모듈로 CSV 열 자르기

종종 쉼표로 구분된 값(CSV) 파일을 통해 구문 분석하고 특정 데이터 열을 추출합니다. Python의 csv 모듈을 사용하면 이 작업을 쉽게 수행할 수 있습니다.

열 추출

특정 열을 추출하려면 열 인덱스를 지정하는 include_cols 변수를 활용하세요. 포함됩니다. 각 열은 0부터 시작하여 번호가 지정되며 첫 번째 열은 0입니다. 예를 들어 "ID", "이름", "우편번호" 및 "전화" 열을 캡처하려면 include_cols = [1, 2, 6을 작성합니다. , 7].

반복 및 인쇄

행을 반복하고 지정된 열을 추출하려면 다음을 사용합니다. for 루프 내의 목록 이해. 각 행에 대해 콘텐츠 변수는 지정된 열의 값으로 채워집니다. 추출된 데이터를 표시하려면 루프 내에서 print(content)를 사용하세요.

included_cols = [1, 2, 6, 7]

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)

Pandas로 최적화

보다 효율적인 CSV 처리를 위해서는 pandas 모듈 사용을 고려하세요. . Pandas는 CSV 파일을 읽고 조작하기 위한 편리하고 강력한 도구를 제공합니다. CSV를 읽고 특정 열을 변수에 저장하려면 다음을 사용하세요.

import pandas as pd
df = pd.read_csv(csv_file)
saved_column = df.column_name

위 내용은 Python을 사용하여 CSV 파일에서 특정 열을 추출하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!