リストの中身がおかしい

###困っていること
10個の引数を与えて、その数の分ずつ配列に数字を入れる関数を作成したのですが、考えている結果になりません。
単純なミスなのかもしれませんが、教えていただきたいです。
###作成した関数

python
1def set_category(sp_0,sp_1,sp_2,sp_3,sp_4,sp_5,sp_6,sp_7,sp_8,sp_9):
2    category = []
3    for i in range(0,sp_0):
4        category.append(0)
5    for i in range(sp_0,sp_1):
6        category.append(1)
7    for i in range(sp_1,sp_2):
8        category.append(2)
9    for i in range(sp_2,sp_3):
10        category.append(3)
11    for i in range(sp_3,sp_4):
12        category.append(4)
13    for i in range(sp_4,sp_5):
14        category.append(5)
15    for i in range(sp_5,sp_6):
16        category.append(6)
17    for i in range(sp_6,sp_7):
18        category.append(7)
19    for i in range(sp_7,sp_8):
20        category.append(8)
21    for i in range(sp_8,sp_9):
22        category.append(9)
23    return category

###失敗例

python
1category = []
2category = set_category(11,3,8,5,8,7,9,4,6,11)

このようにした場合、

python
1category = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 2, 2, 2, 2, 4, 4, 4, 6, 6, 8, 8, 9, 9, 9, 9, 9]

となっていしまいます。
理想は

python
1category = [0,0,0,0,0,0,0,0,0,0,0,1,1,1,2,2,2,2,2,2,2,2,3,3,3,3,3,4,4,4,4,4,4,4,4,5,5,5,5,5,5,5,6,6,6,6,6,6,6,6,6,7,7,7,7,8,8,8,8,8,8,9,9,9,9,9,9,9,9,9,9,9]

このような形です。

ozwk

2017/12/19 06:32

ほしい結果を書いてください

aoisj

2017/12/19 06:41

修正依頼ありがとうございます。質問文の方に追加させていただきました。

行動規範の内容に同意します

回答1件

ベストアンサー

category = set_category(11,3,8,5,8,7,9,4,6,11)で
for i in range(0,sp_0):だと、0 → sp_0=11なので[0,0,...]:11個
for i in range(sp_0,sp_1):だと、sp_0=11 → sp_1=3なので[]:0個
for i in range(sp_1,sp_2):だと、sp_0=3 → sp_1=8なので[2,2,...]:5個
...
となります。

python
1a = (11,3,8,5,8,7,9,4,6,11)
2ans = [[i for k in range(j)] for i,j in enumerate(a)]
3ans = [i for v in ans for i in v]
4print(ans)

python
1def set_category(a):
2    ans = [[i for k in range(j)] for i,j in enumerate(a)]
3    return [i for v in ans for i in v]
4a = (11,3,8,5,8,7,9,4,6,11)
5ans = set_category(a)
6print(ans)

python
1def set_category(a):
2    ans = []
3    for i,j in enumerate(a):
4        ans += [i for k in range(j)]
5    return ans
6a = (11,3,8,5,8,7,9,4,6,11)
7ans = set_category(a)
8print(ans)