CSV 모듈과 Pandas를 사용하여 CSV 파일에서 특정 열을 읽는 방법은 무엇입니까?

CSV 모듈을 사용하여 CSV 파일에서 특정 열 읽기: 종합 가이드

CSV 파일을 구문 분석하고 특정 열에서 데이터를 추출하려는 욕구 열은 데이터 분석의 일반적인 작업입니다. 이 주제를 자세히 알아보기 위해 예시 CSV 파일을 살펴보겠습니다.

ID | Name | Address | City | State | Zip | Phone | OPEID | IPEDS |
10 | C... | 130 W.. | Mo.. | AL... | 3.. | 334.. | 01023 | 10063 |

ID, 이름, 우편번호 및 전화번호가 포함된 열만 캡처해야 한다고 가정해 보겠습니다.

사용 CSV 모듈

초기 접근 방식은 다음을 사용하여 각 행을 반복하는 것이었습니다. 행[열_번호]. 그러나 이 방법은 효과가 없는 것으로 판명되었습니다. 대신, CSV 모듈의 리더 메소드를 사용하여 원하는 열을 지정할 수 있습니다.

import csv

included_cols = [1, 2, 6, 7]
with open(csv_file, 'rb') as csvfile:
    reader = csv.reader(csvfile, delimiter=' ')
    for row in reader:
        content = list(row[i] for i in included_cols)
        # Print the specific columns for each row
        print(content)

이 코드는 각 행에 대해 원하는 열을 인쇄합니다.

Pandas 소개

위 방법은 기능적이지만 Pandas 라이브러리는 CSV 파일 작업을 위한 보다 우아한 솔루션을 제공합니다. Pandas를 사용하면 CSV 파일을 읽고 특정 열을 변수에 저장하는 것이 간단합니다.

import pandas as pd

# Read the CSV file into a DataFrame
df = pd.read_csv(csv_file)

# Save a specific column into a variable
names = df['Name']

결론

다음을 사용하여 CSV 파일에서 특정 열을 읽으려면 CSV 모듈에서는 행을 반복하고 목록 이해를 사용하여 원하는 열을 추출합니다. 보다 포괄적인 솔루션을 위해서는 CSV 파일 조작을 위해 사용하기 쉬운 API를 제공하는 Pandas 라이브러리를 사용하는 것이 좋습니다.

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