PythonでCSVファイルを読み込み、要素を配列に格納する方法

Pythonはその豊富なライブラリとシンプルな構文で、データ解析の分野で非常に人気のあるプログラミング言語です。一方、CSV (Comma Separated Values)ファイルはそのシンプルさと普遍性から、多くのプログラムでデータの保存や交換のために用いられます。PythonとCSVを組み合わせることで、データの読み込み、書き出し、解析などを効率的に行えます。

CSVファイルの読み込み

Pythonのcsvモジュールには、csvファイルを読み込むためのcsv.readerという関数が用意されています。この関数を使用して、CSVファイルを読み込む方法を見てみましょう。

import csv

with open('sample.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

このコードを実行すると、CSVファイルの各行がリストとして出力されます。リストの各要素は、それぞれCSVファイルの列に対応します。

要素の取得

行や列、要素を指定して取得するには、リストのインデックスを指定してください。

# 特定の行を取得する場合
target_row_index = 2  # 取得したい行のインデックス（0から始まる）

with open('sample.csv', 'r') as f:
    reader = csv.reader(f)
    for i, row in enumerate(reader):
        if i == target_row_index:
            print(row)

# 特定の列を取得する場合
target_column_index = 1  # 取得したい列のインデックス（0から始まる）

with open('sample.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row[target_column_index])

CSVファイルの内容をリストや辞書に格納

CSV形式のデータなので、列 (カラム)ごとに個別のデータとして扱えると処理がしやすいですね。csv.reader()とfor文を組み合わせることで、行単位でデータを配列 (list)に格納できます。

import csv

data = []
with open('sample.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        data.append(row)

print(data)

このようにPythonを使ってCSVファイルを読み込み、要素を配列に格納することができます。これにより、データ解析やデータ処理を効率的に行うことが可能になります。.