Pythonで配列内の各要素をcsvファイルに出力する方法

Question

import csv

HEADER = ['果物','特徴一覧']

with open('果物の特徴.csv', 'w', encoding='utf-8') as f:

    writer = csv.writer(f)
    writer.writerow(HEADER)

    fruit = 'りんご'
    feature = ['120円','赤い','丸い']

    row = [fruit,feature]
    writer.writerow(row)

このコードを実行したときにcsvファイルに出力される結果が以下のようになってしまいます。

---

果物,特徴一覧
りんご,"['120円', '赤い', '丸い']"

---

上の出力結果を次のようにしたいと思っています。

---

果物,特徴一覧
'りんご','120円', '赤い', '丸い'

---

この時にfeature内の要素数がn個の時でも対応できる方法を知りたいです。

Answer 1

quoting=csv.QUOTE_ALL を指定していますのでカラム名もシングルクォートで囲まれます。

import csv

HEADER = ['果物','特徴一覧']

with open('果物の特徴.csv', 'w', encoding='utf-8') as f:

    writer = csv.writer(f, quotechar="'", quoting=csv.QUOTE_ALL)
    writer.writerow(HEADER)

    fruit = 'りんご'
    feature = ['120円','赤い','丸い']

    #row = [fruit,feature]
    row = [fruit] + feature
    writer.writerow(row)

果物の特徴.csv

'果物','特徴一覧'
'りんご','120円','赤い','丸い'

Comments

[ddr]

詳細な回答ありがとうございました。

Answer 2

アンパックを活用すればよいかと思います。

row = [fruit, *feature]

Comments

[ddr]

すごくシンプルな解決策でした。ありがとうございました。

Answer 3

質問文にある

果物,特徴一覧
'りんご','120円', '赤い', '丸い'
では、データ行は４つになってしまいますね。

'りんご','120円, 赤い, 丸い'
のようにすれば、データ行も２列になります。

このように出力するには次のようにします。
p.py

import csv

HEADER = ['果物','特徴一覧']
DATA = [
  { 'fruits':  'りんご', 'feature': ['120円','赤い','丸い'] }
]

with open('果物の特徴.csv', 'w', encoding='utf-8') as f:
  writer = csv.writer(f, quotechar="'", quoting=csv.QUOTE_ALL)
  writer.writerow(HEADER)

  for x in DATA:
    row = [x['fruits'], ','.join(x['feature'])]
    writer.writerow(row)

実行例：

参考情報

• https://atmarkit.itmedia.co.jp/ait/articles/2109/28/news020.html CSVファイルの読み書きまとめ