既存のcsvファイルの右に、math関数を使って計算した値の列を加えたいです

Question

実現したいこと

こちらの関連質問です。

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

x
11
21
31

今、y=x/6.1、z=log10(y)として計算した値をx列の隣に追加したいです。「該当のソースコード」の7行目までは正しく回ります。どなたかコードを直してください。

欲しい結果（result.csv）は以下の通りです。

x	y	z
11	1.803278689	0.25606285
21	3.442622951	0.53688946
31	5.081967213	0.706031859

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

Traceback (most recent call last):
  File "〇〇〇\testCV658teratail2.py", line 13, in <module>
    dg['z'] = math.log10(dg['y'])
              ~~~~~~~~~~^^^^^^^^^
  File "〇〇〇\site-packages\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

dg = pd.read_csv('data.csv')
x1 = dg['x']

dg['y'] = x1/6.1
dg['z'] = math.log10(dg['y'])

filename = 'result.csv'
dg.to_csv(filename, sep=',', index=False)

試したこと

「該当のソースコード」の8行目を

dg['z'] = math.log10(float(dg['y']))

にしても同様のエラーが出てしまいました。

Answer 1

以前の質問と同様に numpy.log10() を利用します。(ベクトル演算)

import pandas as pd
import numpy as np
import math

dg = pd.read_csv('data.csv')
x1 = dg['x']

dg['y'] = x1/6.1
dg['z'] = np.log10(dg['y'])

filename = 'result.csv'
dg.to_csv(filename, sep=',', index=False)

math.log10 を使いたい場合は pandas.Series.apply() でよいかと思います。(forループによる処理と同等です)

import pandas as pd
import math

dg = pd.read_csv('data.csv')
x1 = dg['x']

dg['y'] = x1/6.1
dg['z'] = dg['y'].apply(math.log10)

filename = 'result.csv'
dg.to_csv(filename, sep=',', index=False)

Comments

[yyicp]

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