列数が未知のDataFrameでの合計を計算したい

Question

前提・実現したいこと

次のようなCSVファイルが複数あり、アルファベットが振られた列名は左端のAの項目以外、CSVファイルによってまちまちで数十項目あります。

CSVその1
|Aの量|Aの項目1|Aの項目2|Bの量|Bの項目1|Tの量|Tの項目1|Tの項目2|Tの項目3|
|:--|:--:|--:|
| |||

CSVその2
|Aの量|Aの項目1|Aの項目2|Eの量|Gの量|Gの項目1|
|:--|:--:|--:|
| |||

このようなCSVファイルについて、それぞれの項目の割合を計算しようとしています。

CSVその1計算後
|Aの量|Aの項目1|...|Aの量の割合|Bの量の割合|Tの量の割合|
|:--|:--:|--:|
| | | | | | | 100xA/(A+B+T) | 100xB/(A+B+T) | 100xT/(A+B+T) |

該当のソースコード

考えついたのは列を読み込むたびに例外処理を行っていくというものです。
実際にはCSVファイルは複数あるのでosモジュールでCSVファイル名のリストを取得し、for文を用いてファイル名を次々代入して処理していますが省略しています。

import pandas as pd

df = pd.read_csv("hoge.csv")
vol_a = df["Aの量"]
# 項目ごとに量を取り出す
vol_sum = vol_a
try:
    vol_b = df["Bの量"]
    vol_sum += vol_b
except:
    pass
try:
    vol_c = df["Cの量"]
    vol_sum += vol_c
except:
    pass
(以下省略)
# 各量の割合を計算する
df["Aの量の割合"] = vol_a / vol_sum
try:
    df["Bの量の割合"] == vol_b / vol_sum
except:
   pass
(以下省略)

しかし、列の項目数が全部で100近くあり、できればもっと簡潔に書けないかと考えています。

試したこと

1. 複数の処理にまとめて例外処理を行う
2. 条件に一致する（列名に「量」を含む）列だけを取り出して計算する

のどちらかができればと考えているのですが、どちらの方法についても検索しても目的の内容がヒットしませんでした。
上記の内容に限らず、上手い書き方がありましたらご教示いただければ幸いです。

補足情報（FW/ツールのバージョンなど）

Python 3.8.5
Pandas 1.2.0

Answer 1

こういう出力であっていますか。

>>> import pandas as pd
>>> import io
>>> indata = '''Aの量 Aの項目1 Aの項目2 Bの量 Bの項目1 Tの量 Tの項目1 Tの項目2 Tの項目3
... 1 2 3 4 5 6 7 8 9
... 11 12 13 14 15 16 17 18 19
... 21 22 23 24 25 26 27 28 29'''
>>> with io.StringIO(indata) as f:
...     df = pd.read_csv(f, sep=' ')
...
>>> print(df)
   Aの量  Aの項目1  Aの項目2  Bの量  Bの項目1  Tの量  Tの項目1  Tの項目2  Tの項目3
0    1      2      3    4      5    6      7      8      9
1   11     12     13   14     15   16     17     18     19
2   21     22     23   24     25   26     27     28     29
>>> quantities = [x for x in df.columns if x[-2:] == "の量"]
>>> df_sum = df[quantities].sum(axis=1)
>>> for x in quantities:
...     df[x+'の割合'] = df[x]/df_sum
...
>>> print(df)
   Aの量  Aの項目1  Aの項目2  Bの量  Bの項目1  Tの量  Tの項目1  Tの項目2  Tの項目3    Aの量の割合    Bの量の割合    Tの 量の割合
0    1      2      3    4      5    6      7      8      9  0.090909  0.363636  0.545455
1   11     12     13   14     15   16     17     18     19  0.268293  0.341463  0.390244
2   21     22     23   24     25   26     27     28     29  0.295775  0.338028  0.366197

Comments

[退会済みユーザー]

なるほど！
・df.columnsで列名をリストにする
・順番にif文で"の量"が含まれる列名だけのリストを作る
・列名で絞り込みしてdf.sum()で合計を取得
・"の量"が含まれる列名のリストを基に順番に割合を計算
ということですね。現状よりもずっと楽に書くことができそうです。大変ありがとうございます。