アンケート回答の空セルの置換方法

Question

前提

Pandasでアンケートデータの集計をしています。
回答者（行）ごとに選択肢回答を確認して、いずれかの選択肢に回答がある場合に、空白セルを文字列の０（ゼロ）に置換したいと考えています。
質問や選択肢が増えても、同じコードで処理できる状態を実現したいと考えています。

実現したいこと

以下、インデクスをユーザーID、カラム名を質問-選択肢として表現しています。
また、回答がある場合には文字列の1、回答がない場合には空白が入っています。
Q1は必須回答、Q2とQ3は任意回答のイメージです。

上記のように回答者IDごとに、Q1～Q3の回答がある場合、
それぞれの質問において、いずれかの選択肢に回答がある行については、回答のない空白セルを0に置換します。

ただし、Q2-1～Q2-3の回答者1の行や、Q3-1～Q3-2の回答者2の行のように、それぞれの質問において全く回答がない場合は、0置換せずに空白のままとします。

完成イメージは以下の通りです。

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

環境
・Windows10
・conda 4.10.3
・python 3.10.2

Answer 1

import pandas as pd
import io

csv_data = '''
ID,Q1-1,Q1-2,Q2-1,Q2-2,Q2-3,Q3-1,Q3-2
1,1, , , , , ,1
2,1, ,1, ,1, , 
3, ,1, , ,1,1, 
4, ,1, , ,1, , 
5,1, , , , ,1, 
6,1, , , ,1,1, 
'''
df = pd.read_csv(io.StringIO(csv_data))

#
required = ['Q1']
optional = ['Q2', 'Q3']

# required
cols = df.filter(regex=rf'^({"|".join(required)})-').columns
df[cols] = df[cols].replace(' ', 0)

# optional
for c in optional:
  cols = df.filter(regex=rf'^{c}-').columns
  df[cols] = df[cols].apply(lambda x:
    x.replace(' ', 0) if (x=='1').any() else x, axis=1)

print(df.to_markdown(index=False))

ID	Q1-1	Q1-2	Q2-1	Q2-2	Q2-3	Q3-1	Q3-2
1	1	0				0	1
2	1	0	1	0	1
3	0	1	0	0	1	1	0
4	0	1	0	0	1
5	1	0				1	0
6	1	0	0	0	1	1	0

Comments

[aruma_]

勉強になります。ありがとうございました。