appendで追加した作成した配列をソートする

for文内でappendを用いて情報をまとめた後にそれらをある要素でソートしたいのですがsorted関数での指定の仕方が分かりません。そもそもappendで追加されるものは文字列なイメージがあるのでやり方が異なる気もします。アドバイスいただけると幸いです。

現在のコード
Info = list()
for x in len(data)
---Info.append({
------"label": label,
------"confidence": score,
------"topleft": {
---------"x": left,
---------"y": top},
------"bottomright": {
------"x": right,
------"y": bottom}
---}) % 代入される値はforループの中で取得しています
Info = sorted(boxesInfo, key = lambda x: x[3][1]) % "bottomright""y"でソートをしたい

現在のエラー
boxesInfo = sorted(boxesInfo, key = lambda x: x[3][1])
KeyError: 3

opyon

2018/09/21 16:46

サンプルデータ→入力→処理→出力→欲しい結果　がコピペするだけで使えるように提示されないと回答しづらいと思いますよ。

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

回答1件

ベストアンサー

コードはコードブロックで囲んでください。

python
1import numpy as np
2from pprint import pprint
3
4def generate_data():
5    label = np.random.randint(10)
6    score = np.random.rand()
7    left, top, bottom, right = np.random.randint(0, 10, 4)
8    
9    sample = {
10        'label': label,
11        'confidence': score,
12        'tl': {"x": left, "y": top},
13        'br': {"x": bottom, "y": right}
14    }
15    return sample
16
17data = []
18for i in range(10):
19    data.append(generate_data())
20
21def print_data(data):
22    for sample in data:
23        print('({}, {})'.format(
24            sample['br']['x'], sample['br']['y']), end=' ')
25    print()
26
27print_data(data)  # ソート前
28data = sorted(data, key = lambda sample: (sample['br']['y'], sample['br']['x']))
29print_data(data)  # ソート後

y-x 順、つまり bottomright 順でソートしています。

(4, 3) (3, 8) (0, 5) (8, 4) (8, 6) (6, 7) (1, 0) (8, 1) (1, 1) (5, 9) 
↓
(1, 0) (1, 1) (8, 1) (4, 3) (8, 4) (0, 5) (8, 6) (6, 7) (3, 8) (5, 9)

昇順ソートしたい場合は、ソートした結果を反転させます。

data = data[::-1]
print_data(data)  # リバース後

(1, 9) (3, 7) (8, 6) (7, 6) (5, 6) (7, 4) (2, 2) (9, 1) (3, 0) (1, 0)

ソートできる仕組みは数値のタプル同士で比較演算が行える事実に基づきます。

python
1# t1[0] < t2[0] の場合
2print((2, 5) < (5, 3))  # True
3# t1[0] > t2[0] の場合
4print((2, 5) < (1, 2))  # False
5# t1[0] == t2[0] かつ t1[1] < t2[1]
6print((2, 2) < (2, 5))  # True

投稿2018/09/21 17:11

編集2018/09/21 17:55

tiitoi

総合スコア21956

Nao1117

2018/09/21 17:52

回答ありがとうございます。申し訳ありません。言葉足らずでした。 brの降順に従ってInfo全体を並び替えたいのです。 label: a br:x:3,y:4 → label: b br:x:5,y:1 label: b br:x:5,y:1 → label: a br:x:3,y:4 label: c br:x:2,y:9 → label: c br:x:2,y:9 で出力はlistのままだと嬉しいです。