PythonでCSVファイルをマスター：読み込み・書き込みの基本から応用まで

名前,年齢,都市
Alice,30,東京
Bob,25,大阪
Charlie,35,名古屋

import csv

# 読み込むCSVファイル名
file_name = "sample_data.csv"

# テスト用のCSVファイルを作成（一時的に）
with open(file_name, 'w', newline='', encoding='utf-8') as f:
    writer = csv.writer(f)
    writer.writerow(["名前", "年齢", "都市"])
    writer.writerow(["Alice", 30, "東京"])
    writer.writerow(["Bob", 25, "大阪"])

# CSVファイルを読み込み
data = []
with open(file_name, 'r', newline='', encoding='utf-8') as f:
    reader = csv.reader(f)
    header = next(reader) # ヘッダー行を読み飛ばす（または変数に格納）
    for row in reader:
        data.append(row)

print("ヘッダー:", header)
print("読み込んだデータ:", data)
# 出力例:
# ヘッダー: ['名前', '年齢', '都市']
# 読み込んだデータ: [['Alice', '30', '東京'], ['Bob', '25', '大阪']]

import csv

# 読み込むCSVファイル名
file_name = "sample_data.csv" # 上記で作成したファイルを使用

# CSVファイルを読み込み（辞書形式）
data_dict = []
with open(file_name, 'r', newline='', encoding='utf-8') as f:
    reader = csv.DictReader(f)
    for row in reader:
        data_dict.append(row)

print("辞書形式で読み込んだデータ:", data_dict)
# 出力例:
# 辞書形式で読み込んだデータ: [{'名前': 'Alice', '年齢': '30', '都市': '東京'}, {'名前': 'Bob', '年齢': '25', '都市': '大阪'}]

# 特定の要素にアクセス
print(data_dict[0]['名前']) # Alice

import csv

# 書き込むデータ
data_to_write = [
    ["商品名", "価格", "在庫数"],
    ["りんご", 150, 100],
    ["みかん", 120, 200],
    ["バナナ", 100, 150]
]

# 出力CSVファイル名
output_file = "products.csv"

# CSVファイルに書き込み
with open(output_file, 'w', newline='', encoding='utf-8') as f:
    writer = csv.writer(f)
    writer.writerows(data_to_write) # 複数の行を一度に書き込む

print(f"'{output_file}' にデータを書き込みました。")

# 書き込んだファイルの中身を確認 (おまけ)
with open(output_file, 'r', encoding='utf-8') as f:
    print("\n--- 書き込んだファイルの内容 ---")
    print(f.read())

import csv

# 書き込むデータ（辞書のリスト）
data_dict_to_write = [
    {"名前": "田中", "年齢": 40, "部署": "営業"},
    {"名前": "佐藤", "年齢": 32, "部署": "開発"},
    {"名前": "鈴木", "年齢": 28, "部署": "広報"}
]

# フィールド名（ヘッダーとして使用）
fieldnames = ["名前", "年齢", "部署"]

# 出力CSVファイル名
output_file_dict = "employees.csv"

# CSVファイルに書き込み
with open(output_file_dict, 'w', newline='', encoding='utf-8') as f:
    writer = csv.DictWriter(f, fieldnames=fieldnames)
    
    writer.writeheader() # ヘッダー行を書き込む
    writer.writerows(data_dict_to_write) # データを書き込む

print(f"'{output_file_dict}' にデータを書き込みました。")

# 書き込んだファイルの中身を確認 (おまけ)
with open(output_file_dict, 'r', encoding='utf-8') as f:
    print("\n--- 書き込んだファイルの内容 ---")
    print(f.read())

import pandas as pd

# 読み込むCSVファイル名（上記で作成したproducts.csvを使用）
file_name_pd = "products.csv"

# CSVファイルをDataFrameとして読み込み
df = pd.read_csv(file_name_pd)

print("pandasで読み込んだDataFrame:\n", df)

# 出力例:
# pandasで読み込んだDataFrame:
#    商品名  価格  在庫数
# 0   りんご  150  100
# 1   みかん  120  200
# 2   バナナ  100  150

# 特定の列にアクセス
print("\n'商品名' 列:\n", df['商品名'])

import pandas as pd

# 新しいDataFrameを作成
data_for_df = {
    '国': ['日本', 'アメリカ', '中国'],
    '人口(万人)': [12500, 33000, 140000],
    '首都': ['東京', 'ワシントンD.C.', '北京']
}
df_new = pd.DataFrame(data_for_df)

# 出力CSVファイル名
output_file_pd = "countries.csv"

# DataFrameをCSVファイルに書き込み
# index=False: DataFrameのインデックスをCSVに書き込まない
df_new.to_csv(output_file_pd, index=False, encoding='utf-8')

print(f"'{output_file_pd}' にDataFrameを書き込みました。")

# 書き込んだファイルの中身を確認 (おまけ)
with open(output_file_pd, 'r', encoding='utf-8') as f:
    print("\n--- 書き込んだファイルの内容 ---")
    print(f.read())