ファイルを読み込んで、その平方根を取るとエラーになります

Question

実現したいこと

example.csvは以下のようなファイルです。

y
2
3

今、z=yの平方根として、yの列の横にzの列を追加したいと思っています。
つまり、欲しい出力は以下の通りです。

y	z
2	1.414213562
3	1.732050808

「該当のソースコード」を実行するとエラーメッセージが出てしまいました。どなたかコードを直してください。

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

Traceback (most recent call last):
  File "〇〇〇\testCV657-2.py", line 10, in <module>
    df['z'] = math.sqrt(df['y'])
              ~~~~~~~~~^^^^^^^^^
  File "〇〇〇\pandas\core\series.py", line 251, in wrapper
    raise TypeError(f"cannot convert the series to {converter}")
TypeError: cannot convert the series to <class 'float'>

該当のソースコード

import pandas as pd
import math

df = pd.read_csv('example.csv')
df['z'] = math.sqrt(df['y'])
print(df)

試したこと

5行目を

df['z'] = df['y']

とすると、正しく回りました。sqrtをとるとエラーになることが分かりました。

エラーメッセージで検索しましたが、よく分かりませんでした。

5行目を

df['z'] = math.sqrt(float(df['y']))

としても同様のエラーが表示されました。

Answer 1

こうしたい、って話?

import pandas as pd
import math

df = pd.read_csv('example.csv')
df['z'] = [math.sqrt(i) for i in df['y']]
print(df)

df['y']ってのは列(series)であって、数値データじゃない。だからmath.sqrtは直接適用出来ない。
リストにmath.sqrtを直接適用できないでしょ?それとおんなじ。
だからPythonに怒られてるわけ。

こういった場合は「列から一つづつ数値を取り出してはmath.sqrtを個々に適用していく。一番簡単なのはリスト内包表記を使う事だ。その結果をdf['z']へ代入する。

   y         z
0  2  1.414214
1  3  1.732051

おしまい。

Comments

[yyicp]

>df['y']ってのは列(series)であって、数値データじゃない

勉強になりました。
どうもありがとうございました。

Answer 2

この場合にはベクトル演算を行う numpy.sqrt() を使うとよいかと思います。

import numpy as np
import pandas as pd

df = pd.read_csv('example.csv')
df['z'] = np.sqrt(df['y'])
print(df)

#    y         z
# 0  2  1.414214
# 1  3  1.732051

Comments

[yyicp]

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