既存のcsvファイルに、そのcsvファイルから計算した新しい列を追加したい

Question

実現したいこと

data1.csvは次のようなファイルです。

x	y
1	2
3	4

data2.csvは次のようなファイルです。

x	y
5	6
7	8

この2つのファイルを読み込んで、3<y<7を満たすデータだけを取り出して、z=xy+1の標準偏差を計算したいです。

やりたいことを模式的に書くと下図のようになります。

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

Traceback (most recent call last):
  File "…\testCV571.py", line 9, in <module>
    z=x*y+1
      ^
NameError: name 'x' is not defined

該当のソースコード

こちらを参考にして書きました。

import pandas as pd
import statistics
from glob import glob

ymin = 3
ymax = 7

df = pd.concat([pd.read_csv(f) for f in glob("data*csv")])
z=x*y+1
l = df.query("@ymin < `y` < @ymax")["z"].tolist()

print(l)   
print(statistics.stdev(l))

試したこと

9行目の先頭に#をつけ、10行目の["z"]を["y"]にすると、正しく動いて1.4142135623730951が出力されます。

Answer 1

z=x*y+1 をデータフレーム上の計算式に変更します。

import pandas as pd
import statistics
from glob import glob

ymin = 3
ymax = 7

df = pd.concat([pd.read_csv(f) for f in glob("data*csv")])
# z=x*y+1
df["z"] = df["x"] * df["y"] + 1
l = df.query("@ymin < `y` < @ymax")["z"].tolist()

print(l)   
print(statistics.stdev(l))

# [13, 31]
# 12.727922061357855

Comments

[yyicp]

できました。
どうもありがとうございました。