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 ファイルへの行の読み取りと CSV ファイルへの行の書き込みをそれぞれ処理します。
• with ステートメントを備えた open 関数により、ファイルの適切な処理とクリーンアップが保証されます。
• next(reader) 関数は最初の行 (ヘッダー) を読み取り、ヘッダーに新しい列を追加するために必要なイテレータを進めます。
• Python 3 では、イテレータは csv から取得されます。 .reader はタプルを生成するため、変更して書き込む前にタプルをリストに変換する必要があります。

これらの手順に従うことで、CSV ファイルに新しい列を正常に追加し、データ表現を強化できます。

以上がPython を使用して既存のデータから派生した新しい列を CSV ファイルに追加する方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。