Python을 사용하여 기존 데이터에서 파생된 새 열을 CSV 파일에 추가하는 방법은 무엇입니까?

CSV 파일 수정: 새 열 추가

문제 설명

'이름' 및 '코드'라는 열이 포함된 CSV 파일이 주어지면 'Name' 열에서 파생된 값을 사용하여 'Berry'라는 새 열을 추가하는 것을 목표로 합니다. 원하는 출력은 다음과 유사해야 합니다.

Name        Code    Berry
blackberry  1   blackberry
wineberry   2   wineberry
rasberry    1   rasberry
blueberry   1   blueberry
mulberry    2   mulberry

제안된 솔루션

Python과 CSV 모듈을 사용하여 다음과 같이 CSV 파일을 조작할 수 있습니다.

1. 읽기용 입력 CSV 파일과 쓰기용 출력 CSV 파일.
2. csv.reader를 사용하여 입력 파일의 각 행을 반복합니다.
3. 각 행에 대해 새 ' Berry' 열.
4. csv.writer를 사용하여 수정된 행을 출력 파일에 씁니다.

다음은 예제 스크립트입니다.

<code class="python">import csv

with open('input.csv', 'r') as input_file, open('output.csv', 'w') as output_file:
    reader = csv.reader(input_file)
    writer = csv.writer(output_file)

    # Read the header row and add the new column
    header = next(reader)
    header.append('Berry')
    writer.writerow(header)

    # Iterate over the remaining rows and modify them
    for row in reader:
        row.append(row[0])    # Set the 'Berry' column to the 'Name' column
        writer.writerow(row)</code>

주요 고려 사항

• csv.reader 및 csv.writer 함수는 각각 CSV 파일에서 행 읽기 및 쓰기를 처리합니다.
• with 문을 사용한 open 함수는 적절한 파일 처리 및 정리를 보장합니다.
• next(reader) 함수는 첫 번째 행(헤더)을 읽고 헤더에 새 열을 추가하는 데 필요한 반복자를 전진시킵니다.
• Python 3에서는 csv에서 얻은 반복자가 .reader는 튜플을 생성하므로 수정하고 작성하기 전에 이를 목록으로 변환해야 합니다.

이 단계를 따르면 CSV 파일에 새 열을 성공적으로 추가하여 데이터 표현을 향상할 수 있습니다.

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