CSVファイルをitertuplesで行ごとに抽出した値を使って四則演算し、演算結果を列に追加したい

Question

itertuplesで行ごとに抽出した値を使って四則演算したい

コード
import pandas as pd
import numpy as np
import csv

def calc(x):
a = x[0]
b = x[1]
c = x[2]

A = a+b+c
B = a*b*c

return(A,B)


fname = "calc_test"
df_test = pd.read_csv("{}.csv".format(fname))
for row in df_test.itertuples(name=None):
AB = row[1:]
print(AB)
print(calc(AB))

print(df_test)

(2, 2, 1)
(6, 6)
(2, 3, 5)
(6, 6)
(2, 9, 2)
(6, 6)

・csvの中身
a b c
0 2 2 1
1 2 3 5
2 2 9 2

（実際はカンマ区切りになります）

試したこと

calc(bb)を定義したあとに、
変数a, b, cそれぞれの値をcsvから読み込んで、
四則演算（式A、式B）を実行したいと考えています。

実際はもっと大規模になるのですが、
小規模にしたものを上記コードになります。

読み込み自体はできているのですが、
１行目のcalc()、２行目のcalc()、・・・をそれぞれ出力できませんでした。

さらに、この演算結果をD列に追加し、CSVファイルに書き込みを行いたいと考えています。

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

python3.7

Comments

[meg_]

掲載のコードを実行すると下記エラーが出ます。
「NameError: name 'x' is not defined」
エラーの出ないコードを掲載してください。

[nsk]

修正をしました。
ご確認お願い致します。

import pandas as pd
import numpy as np
import csv

def calc(x):
a = x[0]
b = x[1]
c = x[2]

A = a+b+c
B = a*b*c

return(A,B)

fname = "calc_test"
df_test = pd.read_csv("{}.csv".format(fname))
for row in df_test.itertuples(name=None):
AB = row[1:]
print(AB)
print(calc(AB))

print(df_test)

[meg_]

１．掲載のコードを修正してください。（他の人が見られるように）
２．csvファイルの中身はタブ区切り（TSV）のように見えます。実際のファイルの中身はカンマ区切りなのですか？

[can110]

「出力できませんでした」とは具体的にどのような結果になるのでしょうか？
また入力CSVの区切り文字が何か（半角空白２文字でよい？）を記載ください。

Answer 1

さらに、この演算結果をD列に追加し、

「出力されません」の詳細が分かりませんが、以下のような感じでD列を作成できます。

import pandas as pd
from io import StringIO

def calc(x):
    a, b, c = x
    A = a+b+c
    B = a*b*c
    return(A,B)

df = pd.read_csv(StringIO(",a,b,c\n0,2,2,1\n1,2,3,5\n2,2,9,2"), index_col=0)
df['D'] = [calc(row[1:]) for row in df.itertuples()]
print(df)
"""
   a  b  c         D
0  2  2  1    (5, 4)
1  2  3  5  (10, 30)
2  2  9  2  (13, 36)
"""

Comments

[nsk]

ご回答ありがとうございました。
問題を解決することができました。
ありがとうございました。

Answer 2

出力結果は下記となりました。

import pandas as pd
import numpy as np
import csv

def calc(x):
    a = x[0]
    b = x[1]
    c = x[2]

    A = a+b+c
    B = a*b*c

    return(A,B)


fname = "calc_test"
df_test = pd.read_csv("{}.csv".format(fname))

for row in df_test.itertuples(name=None):
    AB = row[1:]
    print(AB)
    print(calc(AB))

【出力結果】
(2, 2, 1)
(5, 4)
(2, 3, 5)
(10, 30)
(2, 9, 2)
(13, 36)

DataFrameへの追加は、calc()の戻り値が２値のタプルのため簡単に２列追加する場合とします。
こんな感じでしょうか。

for row in df_test.itertuples(name=None):
    AB = row[1:]
    result = calc(AB)
    df_test.loc[row[0],'D'] = result[0]
    df_test.loc[row[0],'E'] = result[1]

df_test.to_csv('calc_test2.csv', index=False)

Comments

[meg_]

floatからintへの変換は下記で。
df_test[['D','E']] = df_test[['D','E']].astype(np.int64)

[nsk]

詳しく説明していただき理解することができました。
ありがとうございました。
問題を解決することができました。