python: csvをjsonに変換すると、数値が文字列になってしまいます

Question

前提・実現したいこと

csvデータの内容を若干変更し、jsonに変換したいと思っております。

下記のようなコードにしたところ、idのvalueが文字列となってしまいます。
数値としてjsonデータにするには、どのようにすれば良いかわからず
質問させていただきました。
ご教授いただければ幸いです。何卒よろしくお願い致します。

※元となるcsvでは、サイズが幅（W）と高さ（H）にわかれていますが
jsonでは幅×高さという表記にしたいため、
冗長だと思ったのですが、test2.csvを作成→json変換としました。

発生している問題・エラーメッセージ

// "id"の"1001"を1001としたいです。（1002も）
[
  {
    "id": "1001",
    "name": " テスト1",
    "size": " 50 × 100"
  },
  {
    "id": "1002",
    "name": " テスト2",
    "size": " 60 × 110"
  }
]

該当のソースコード

id,name,sizeW,sizeH
1001, テスト1, 50, 100
1002, テスト2, 60, 110

import json
import csv

json_list = []

#元データを開く（test.csv）
with open('test.csv', mode='r', encoding='utf-8') as rf:
    reader = csv.reader(rf)
    next(reader)
    #サイズ表記をまとめたcsvを作成（test2.csv）
    with open('test2.csv', mode='w', encoding='utf-8') as wf:
        writer = csv.writer(wf)
        writer.writerow(['id', 'name', 'size'])
        for line in reader:
           id = line[0]
           name = line[1]
           sizeW = line[2]
           sizeH = line[3]
           size = sizeW + ' ×' + sizeH
           writer.writerow([id, name, size])

with open('test2.csv', 'r') as f:
    for line in csv.DictReader(f):
        json_list.append(line)

#jsonデータを作成
with open('test.json', 'w') as f:
    json.dump(json_list, f, indent=2, ensure_ascii=False)

Answer 1

もしpandasも使って良いというのでしたら、csvモジュールより楽かと思います。

import pandas as pd

df = pd.read_csv("test.csv")
df["size"] = df["sizeW"].astype(str) + ' × ' + df["sizeH"].astype(str)
df2 = df.loc[:, ["id", "name", "size"]]
print(df2)
#      id   name      size
# 0  1001   テスト1  50 × 100
# 1  1002   テスト2  60 × 110

df2.to_json("test.json",orient='records')

Comments

[hihamo]

ご回答ありがとうございます！
pandasを使ったことがなかったのですが、
こんなにシンプルなコードになるのですね。
インストールしてみます。ありがとうございます。

Answer 2

以下のようにid = int(line[0])とします。
そして読み込んだ結果からそのままjson_listを作成します。
あとはjson_listをdumpすればよいです。

import json
import csv

json_list = []

#元データを開く（test.csv）
with open('test.csv', mode='r', encoding='utf-8') as rf:
    reader = csv.reader(rf)
    next(reader)
    for line in reader:
        id = int(line[0])
        name = line[1]
        sizeW = line[2]
        sizeH = line[3]
        size = sizeW + ' ×' + sizeH
        json_list.append( {'id':id, 'name':name, 'size':size})

#jsonデータを作成
with open('test.json', 'w', encoding='utf-8') as f:
    json.dump(json_list, f, indent=2, ensure_ascii=False)

Comments

[hihamo]

ご回答ありがとうございます！
数値のままjsonに変換できました。

このようにintを使って直接listを作成すれば良いのですね。
間に一旦csvを作成することも気になっていたので
こちらも解決できて嬉しいです。ありがとうございました！

Answer 3

csv.readerを使って読み込んだlineは文字列のリストです。
整数にしたければ組み込みクラスintを使って変換しましょう。

Comments

[hihamo]

ご回答ありがとうございます！
組み込みクラスint・・というのがあるのですね。
調べてみます。ありがとうございました。

Answer 4

ダブルクォーテーションで囲むと文字列として扱われます。
数値として扱いたければダブルクォーテーションを外してください。

↓参考
https://www.tohoho-web.com/ex/json.html

Comments

[hihamo]

ご回答ありがとうございます。

csvからjsonに変換する際に、自動でダブルクォーテーションがついてしまうのですが
これをつかない様にするにはどのようなコードにすれば良いのかわからずに質問させていただきました。

intを使うのかとも思ったのですが、どの段階で指定すればよいのか調べてもわからず
質問が見当違いでしたら申し訳ありませんが、教えていただければありがたいです。
よろしくお願い致します。