他の列を条件にして、1列目を合計したい

Question

Python初心者です。
ネットでいろいろ検索したのですが、いい情報が見つかりませんでした。
目的を達成するようなコードは書けましたが、
これよりいい方法がいくらでもあると思うので、どうか教えてください。

＜やりたいこと＞
・合計したい列はデータフレームの1列目。
・対象行は、指定列目以降全て。
（今回は2列目以降。）
・今はあえてprintにしてありますが、
最終的には以下のようなデータフレームを作るつもりです。
col1 col2 col3
101 0 1111

処理が早いことが優先ですが、
きれいでわかりやすいコードだとうれしいです。
どうぞよろしくお願いいたします。

import pandas as pd
 
df = pd.DataFrame({'cnt': [1, 10, 100, 1000],
                    'col1': [1, 0, 1, 0],
                    'col2': [0, 0, 0, 0],
                    'col3': [1, 1, 1, 1]})
 
names = df.columns.values
for x in range(1, len(df.columns)):
    colname = names[x]
    df2 = df[df[colname] == 1]
    print(colname + ':' + str(df2['cnt'].sum()))

Answer 1

こうゆうことでしょうか？

import pandas as pd

df = pd.DataFrame({'cnt': [1, 10, 100, 1000],
                    'col1': [1, 0, 1, 0],
                    'col2': [0, 0, 0, 0],
                    'col3': [1, 1, 1, 1]})

# 方法１
res1 = df.iloc[:,1:].apply(lambda d: df.cnt[d.astype(bool)].sum())
print(res1)
#col1     101
#col2       0
#col3    1111
#dtype: int64

# 方法２
res2 = df.iloc[:,1:].apply(lambda d: (d*df.cnt).sum())
print(res2)
#col1     101
#col2       0
#col3    1111
#dtype: int64

Comments

[miyu21]

ご回答ありがとうございました。
回答がなく諦めているうちに、他のことに気を取られ確認を怠っていました。

書いていただいた内容で完璧です！
まだラムダ式は使えるようになっていないので、使いながら習得します。

私が書きそうなコードは「方法2」、
でも、真理値ということが後でもわかりやすいのは「方法1」
だと思いました。

今後は、開けている質問はきちんと確認するようにします。
本当にありがとうございました。

Answer 2

Pandasなしだとこう書けます。

from operator import mul


data = {'cnt': [1, 10, 100, 1000],
        'col1': [1, 0, 1, 0],
        'col2': [0, 0, 0, 0],
        'col3': [1, 1, 1, 1]}

columns = ['col1', 'col2', 'col3']
values = [sum(map(mul, data['cnt'], data[col]))
          for col in columns]

print(*columns)
print(*values)

追記

思いつく限りの最速かつ読みやすいコードは以下です。

import numpy as np


cnt = np.array([1, 10, 100, 1000])
mat = np.array([[1, 0, 1, 0],
                [0, 0, 0, 0],
                [1, 1, 1, 1]])
col1, col2, col3 = np.inner(cnt, mat)

print('col1', 'col2', 'col3')
print(col1, col2, col3)

Comments

[miyu21]

ご回答、ありがとうございます。

ただ、やりたいことに書いていますように、
「合計したい列はデータフレームの1列目」なのです。
（データフレームにcsvからデータを読み込むため。）

教えて頂いたコードで自分なりに書いてみたのですが、
結局はコードの行数が多くなってしまっています。
でも、掛け算をするという方法は大変参考になりました。

どなたか、他のやり方も教えていただければ幸いです。
よろしくお願いいたします。