pandasで、2列目以降を合計・カウントして結果を追加する

Question

以下のようなデータフレームがあった際に、
行ごとの合計値と、データの個数を計算して、結果を列として追加したいと考えています。

▼データフレーム

import pandas as pd

# データフレームを作る
df = pd.DataFrame({"name":["tarou", "hanako", "jun"], "2021-10-01":["1", "3", "5"], "2021-10-02":["4", "", "3"], "2021-10-03":["5", "4", "8"]})

▼試してみたコード

# 行ごとの合計値と、行ごとの個数をカウントして、結果を列として追加する
df['sum'] = df.sum(axis=1)
df['count'] = df.count(axis=1)
df

しかし、結果が上手く出力されません。
合計値とカウントをしたいのは、データフレームの2列目以降のデータのみなのですが、
合計値とデータの個数をそれぞれ計算して以下のように出力したい場合のコードを教えて頂けますでしょうか。

▼最終的に出力したいもの

Answer 1

無理やりやるなら以下のようになります。

>>> df = pd.DataFrame({"name":["tarou", "hanako", "jun"], "2021-10-01":["1", "3", "5"], "2021-10-02":["4", "", "3"], "2021-10-03":["5", "4", "8"]})
>>>
>>> def convert(s):
...     try:
...         return int(s)
...     except:
...         return 0
...
>>> df['sum'] = sum([df[c].apply(convert) for c in df.columns[1:4]])
>>> df['count'] = (df.iloc[:,1:4]!='').sum(axis=1)
>>> print(df)
     name 2021-10-01 2021-10-02 2021-10-03  sum  count
0   tarou          1          4          5   10      3
1  hanako          3                     4    7      2
2     jun          5          3          8   16      3

しかし、pandasで空文字を含む数の文字列を扱うことは望ましい使いかたではないので、データフレームを作る段階でもう少し考えたほうが良いと思います。

Comments

[color_8]

ppaulさん
ご丁寧に有難うございました。

ひとまずは解決いたしましたが、おっしゃる通りそもそものデータフレームの段階で改良を検討いたします。
今後ともよろしくお願いいたします。