【python】【pandas】列同士を計算して新たな列を追加したい

Question

実現したいこと

列同士を計算して、新たなに列を追加したい。

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

TypeError: can't multiply sequence by non-int of type 'float'

該当のソースコード

df['合計']=df['数']*df['価格']

試したこと

一度目はエラーなしで動作したが、2回目から上記のエラーが発生。

データ型が「float」になっていることが原因と思い、

 df['価格'].astype(int)

を追加したが、

ValueError: cannot convert float NaN to integer

が発生いたしました。

できれば、少数型で計算したいのですが、できないのしょうか。
できなければint型の変換方法はどのように行えばよろしいでしょうか。
何卒、よろしくお願いいたします。

Answer 1

TypeError: can't multiply sequence by non-int of type 'float'

もしかしたら「数」列が文字列型なのかもしれません。なので、pandas.to_numeric() で数値に変換するとよいかもしれません。

import pandas as pd
import numpy as np

pd.set_option('display.unicode.east_asian_width', True)

df = pd.DataFrame({
  '数': ['1', '2', '3', '4', '5'],
  '価格': [10, 20, 30, 40, np.NaN]
})

# TypeError
# df['合計'] = df['数'] * df['価格']

df['合計'] = pd.to_numeric(df['数'], errors='coerce') * df['価格']

print(df)

Comments

[sagiyama]

ご回答いただきありがとうございます。
エラーが発生せずに動作いたしました。

Answer 2

df.fillna(0).astype('int', errors='ignore')

NaNの部分を0埋めすることで変換できるようになります。
float型でやりたければ.astype以降を消せば問題ないと思います。
追記1:

df['価格'].fillna(0).astype('int', errors='ignore')

価格のみゼロ埋めするようにしましたが、どうでしょうか？

Comments

[sagiyama]

ご回答いただきありがとうございます。
教えていただいたコードを実行するとできました。
しかし、そのあとに
「df['不良合計']=df['不良1']+df['不良2']+df['不良3']」
と続くのですが、こちらが、
「TypeError: can only concatenate str (not "int") to str」
と、エラーが出てしまいました。
また、
「df.fillna(0)」
で実行したところ、
「df['合計']=df['数']*df['価格']」部で
「TypeError: can't multiply sequence by non-int of type 'float'」
がでてしまいます。