辞書から必要な要素を条件分岐で抽出すると期待した結果と異なる

###辞書に格納されている配列をキーで条件分岐して抽出したいが正しく抽出できない。

dict = {1:[[1,1,1],[2,2,2]], 2:[[3,3,3],[4,4,4]], 3:[[5,5,5],[6,6,6]], 5:[[9,9,9],[10,10,10]], 7:[[13,13,13],[14,14,14]]}
辞書内にキーが2であるものが存在するか否かで分岐させ、優先順位の高いキーの配列から抽出するといった作業です。
条件分岐の書き方がまずいようで正しく抽出できないのですが、間違っている点が見当つきません。

以下、書いたコードです。

python
1dict = {1:[[1,1,1],[2,2,2]], 2:[[3,3,3],[4,4,4]], 3:[[5,5,5],[6,6,6]], 
2        5:[[9,9,9],[10,10,10]], 7:[[13,13,13],[14,14,14]]}
3size = 40
4# sizeを40で設定
5# 抽出後期待される結果=> result = [[1,1,1],[2,2,2],[3,3,3],[4,4,4],[5,5,5],[6,6,6],[13,13,13],[14,14,14]]
6
7
8result = [] #抽出したものの登録先
9
10if not 2 in dict.keys(): #キー2が存在しない場合
11
12    #1)キーが1のリストの3つ目の値v[2]をカウント
13    get = dict.get(1, [[0]*3])
14    count = 0
15    for v in get:
16        count = int(count) + int(v[2])
17        
18    if count >= size:
19        result = result + get
20
21    else: #2)キーが3のリストの3つ目の値v[2]をカウント
22        result = result + get
23
24        get = dict.get(3, [[0]*3])
25        for v in get:
26            count = int(count) + int(v[2])
27                
28        if count >= size:
29            result = result + get
30                        
31        else: #3)キーが4のリストの3つ目の値v[2]をカウント
32            result = result + get
33
34            get = dict.get(4, [[0]*3])
35            for v in get:
36                count = int(count) + int(v[2])
37                                    
38            if count >= setai:
39                result = result + get
40                                    
41            else:#5)キーが5または6のリストを追加
42                result = result + get
43
44                if 5 in dict.keys():
45                    get = dict[5]
46                    result = result + get
47                
48                if 6 in dict.keys():
49                    get = dict[6]
50                    result = result + get
51    
52    n = 0    
53    for i in result:
54        if i[0] == 0:
55            del result[n]
56        n = n + 1
57        
58n = 0    
59for i in result:
60    if i[0] == 0:
61        del result[n]
62    n = n + 1
63    
64
65else: #キー2が存在するとき
66    
67    #1)キーが1のリストの3つ目の値v[2]をカウント
68        get = dict.get(1, [[0]*3])
69        count = 0
70        for v in get:
71            count = int(count) + int(v[2])
72
73        if count >= size:
74            result = result + get
75            
76        else: #2)キーが2のリストの3つ目の値v[2]をカウント
77            result = result + get
78
79            get = dict.get(2, [[0]*3])
80            for v in get:
81                count = int(count) + int(v[2])
82                        
83            if count >= size:
84                result = result + get
85            
86            else: #3)キーが3のリストの3つ目の値v[2]をカウント
87                    result = result + get
88
89                    get = dict.get(3,[[0]*3])
90                    for v in get:
91                        count = int(count) + int(v[2])
92                    
93                    if count >= size:
94                        result = result + get
95                            
96                    else: #4)キーが4のリストの3つ目の値v[2]をカウント
97                        result = result + get
98
99                        get = dict.get(4, [[0]*3])
100                        for v in get:
101                            count = int(count) + int(v[2])
102                                            
103                        if count >= size:
104                            result = result + get
105                                                
106                        else: #5)キーが7のリストの3つ目の値v[2]をカウント
107                            result = result + get
108
109                            dict.get(7, [[0]*3])
110                            for v in get:
111                                count =int(count) + int(v[2])
112                                                        
113                            if count >= size:
114                                result = result + get
115                                        
116                            else: #6)キーが5または6のリストを追加
117                                result = result + get
118
119                                if 5 in dict.keys():
120                                    get = dict[5]
121                                    result = result + get
122
123                                if 6 in dict.keys():
124                                    get = dict[6]
125                                    result = result + get
126        n = 0    
127        for i in result:
128            if i[0] == 0:
129                del result[n]
130            n = n + 1
131
132n = 0    
133for i in result:
134    if i[0] == 0:
135        del result[n]
136    n = n + 1
137
138print(result)

##実行結果
期待していた結果はresult >> [[1,1,1],[2,2,2],[3,3,3],[4,4,4],[5,5,5],[6,6,6],[13,13,13],[14,14,14]]
でしたが以下のような結果になりました。

Python
1result >> [[1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4], [5, 5, 5], [6, 6, 6], [9, 9, 9], [10, 10, 10]]

##考えた原因
辞書内にキーが存在しないときKeyErrorを回避するためgetメソッドを使って要素が0の配列を返すようプログラムしました。
これがまずかったのかなと考え、if文など他のKeyError回避策も試しましたがダメでした。
分かる方いらっしゃったらご教授お願いします。

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

回答1件

ベストアンサー

                        else: #5)キーが7のリストの3つ目の値v[2]をカウント
                            result = result + get

                            dict.get(7, [[0]*3])

↑ここの部分を↓のように、dict.get～の戻り値をgetに代入するようにしてください。

                        else: #5)キーが7のリストの3つ目の値v[2]をカウント
                            result = result + get

                            get = dict.get(7, [[0]*3]) # <============= getに代入する。

投稿2020/08/30 08:36