[python][CSV]スペースを含むカラムのみクォートして書き込みたい

Question

python 3.7 を使用して、DBから取得したデータをCSVに書き込んでいく(writer.writerow)処理を実装しています。

出力内容が少々複雑なので、助けていただけると幸いです。

実現したいこと

数字、文字列を各行に含む内容をCSVに出力するのですが、「フィールドに半角スペースが含まれる場合のみ」ダブルクォーテーションで囲んだ以下のような状態で出力したいです。

1234,"test test",3456  <= 1. 半角スペースを含むのでクォート
1235,sample,452        <= 2. quotechar を含まないためクォートなしで出力
7890,5gg6,6789         <= 3. 2.と同様
asge,"rg""hk6",        <= 4. quotechar のデフォルト(")に該当するためクォートされる

csvモジュールの writer には delimiter や quotechar があり、それらで出力内容のクォートを制御できるという点に関しては把握していますが、
上記のような出力を実現できる方法はあるでしょうか？

以下、現在の該当部分のソースとなります。

with open("/tmp/" + file_name, "a", encoding="utf-8") as csv_file:
    writer = csv.writer(csv_file)
    # set header
    writer.writerow(header)
    ## Get total number of SELECT data
    with conn.cursor() as cur:
        for i in range(offset, current_block_fetch, limit):
            # get records each 1,000
            cur.execute(query + " LIMIT %s OFFSET %s", [limit, i])
                    
            # get and shape of all selected data
            
            for row in cur.fetchall():
                writer.writerow(row)
            

その他、必要な情報などあればコメント等いただければ適宜追記致します。

注釈：現在はcsvモジュールでの書き込みとなっておりますが、実現できるのであればcsvモジュールを使用しない方法でも問題ありません。

よろしくお願い致します。

Comments

[tiitoi]

quoting に指定できるのは下記4種類だけなので、「空白を含む場合のみ」というオプションは csv モジュールにはないですね。
https://docs.python.org/ja/3/library/csv.html#csv.QUOTE_ALL

どうしてもやりたければ、csv モジュールを使わないで自分で書く必要があると思います。

[zvub1123]

ありがとうございます。やはり作りこむ必要がるようですね。。。

Answer 1

df に該当データが入っているとして

for i in range(len(df.columns)):
  for j in range(len(df)):
    val = str(df.iloc[j, i])
    if '"' in val:
      val = '"' + val.replace('"', '""') + '"'
      df.iloc[j, i] = val
    if ' ' in val:
      val = '"' + val + '"'
      df.iloc[j, i] = val
df.to_csv('out.csv')

こんな感じでしょうか

Comments

[zvub1123]

ありがとうございます！Pandasを使うとスマートにできそうですね。