標準入力とグループ分けの問題で、コードを簡易化できないか考えています。
①A,B,C,D,Eの5人を標準入力から受け取ったデータによって3つのグループに分けます。
※例えば[1 1 3]と入力したら1人、1人、3人の3つのグループになります。
[入力] 2 1 2
[コード] person = ["A","B","C","D","E"] #グループメンバー nums = input().rstrip().split(" ") #標準入力から各グループの人数を受け取る group1 = int(nums[0]) #グループ1は2人 group2 = int(nums[1]) #グループ2は1人 group3 = int(nums[2]) #グループ3は2人 for i in range(group1): #グループ1のメンバーを取り出す print(person[i],end="") print() del person[0:group1] for j in range(group2): #グループ2のメンバーを取り出す print(person[j],end="") print() del person[0:group2] for k in range(group3): #グループ3のメンバーを取り出す print(person[k],end="")
[出力] AB C DE
なんとかコードを書いて欲しい出力は得られたのですが、どうもコードが冗長になっている気がします。
改善したほうが良い部分、変更したほうがいい部分がありましたら教えていただけないでしょうか?
よろしくお願いいたします。
回答3件
あなたの回答
tips
プレビュー