for文の出力結果をcsvファイルに改行して、１行１行全て順番に出したい

Question

前提・実現したいこと

for文の出力結果をcsvファイルに改行して、１行１行全て順番に出したい

ここに質問の内容を詳しく書いてください。
for文で出力結果をcsvに書き込みたいのですが、どのようにしたらよいでしょうか。
下記のコードのsumの一個一個の結果を改行してcsvファイルに書き込みたいです。
よろしくお願いいたします。

■■な機能を実装中に以下のエラーメッセージが発生しました。

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

エラーメッセージ

該当のソースコード

# python 3.6

i = 0
k = 0
n = 100
l = 100
for k in range (l):
    for i in range (n):
        distance = np.linalg.norm(X_2[k:]-X[i:])
        sum = sum + distance
    
    print(sum)
    sum = 0

試したこと

import csv して、writecsvとか色々やってみた
ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

sum.py

import csv

n = 3
l = 3
sum = 0
filename = "sum.csv"
with open(filename, 'w', newline="") as f:
    fieldnames = ["k", "sum"]
    writer = csv.DictWriter(f, fieldnames=fieldnames)
    writer.writeheader()

    for k in range(l):
        for i in range(n):
            # distance = np.linalg.norm(X_2[k:]-X[i:])
            distance = k * 100 + i
            sum = sum + distance

        row = {"k": k, "sum": sum}
        writer.writerow(row)
        sum = 0

実行例

参考情報

• [Python] csvやtsvの読み書きを行う（csvモジュールの利用）

https://www.yoheim.net/blog.php?q=20171003

• CSVファイルの読み書き

https://www.python-izm.com/advanced/csv/

• PythonでCSVの読み書きを行う方法【初心者向け】

https://techacademy.jp/magazine/15638

Comments

[MOSMOS2]

ありがとうございます！

Answer 2

printの出力先を書き出し用のファイルにしてあげれば行けますよ。

print(sum)

を

with open("tmp.csv", "a") as file:
    print(sum, file=file)

にする。

もっとも、現在のコードでも標準出力をファイルにリダイレクトすればターミナル上で書き出させることも出来ます。

python calc_distance.py > tmp.csv

として吐き出すことも出来ます。用途によりけりですが。

Comments

[tachikoma]

sumはPythonの予約語の一つなので、変数名に使うのは極力避けたほうがいいです。

[MOSMOS2]

ありがとうございます！

Answer 3

csv.writer.writerowかcsv.writer.writerowsを使えば問題なくできるでしょう。
前者はcsvの行に対応するリストを一度の呼び出しで1つずつ渡してあげれば使えます。後者はcsvに対応する二重リストを受け取ります。

参考：
PythonでCSVの読み書き - Qiita

Comments

[MOSMOS2]

ありがとうございます！