Добавление различных значений списка в словарь в python - PullRequest
0 голосов
/ 25 декабря 2018

У меня есть три списка, содержащие разные шаблоны значений.Это должно добавить определенные значения только внутри одного словаря на основе некоторых условий if.I попытался сделать это следующим образом, но я получил все значения из списка.

class_list = [1,2,3,4,5,6]

boxes = [[0.1,0.2,0.3,0.4],[0.5,0.7,0.8,0.9],[0.7,0.9,0.4,0.2],[0.9,0.7,0.6,0.3],[0.9,0.14,0.6,0.3],[0.9,0.7,0.6,0.13]]

scores = [0.98,0.87,0.97,0.96,0.94,0.92]

k=1;

data = {}

for a in scores:
    for b in boxes:
        for c in list:
            if a >= 0.98:
                data[k+1] = {"score":a,"box":b, "class": c } ;
                k=k+1;
print("Final_result",data)

Ответы [ 2 ]

0 голосов
/ 25 декабря 2018

Вы получаете все значения, потому что вы перебираете все значения со всеми другими значениями, то есть 6x6x6 раз.Я понял вашу проблему, вам нужно использовать функцию Zip.

llist = [1,2,3,4,5,6]

boxes = [[0.1,0.2,0.3,0.4],[0.5,0.7,0.8,0.9],[0.7,0.9,0.4,0.2],[0.9,0.7,0.6,0.3],[0.9,0.14,0.6,0.3],[0.9,0.7,0.6,0.13]]

scores = [0.98,0.87,0.97,0.96,0.94,0.92]

k=1;

data = {}



for a,b,c in zip(scores, boxes, llist):
    if a >= 0.98:
        data[k] = {"score":a,"box":b, "class": c } ;
        k = k + 1

for k in data.keys():
    print(k , data[k])

ps: избегайте именования переменных с помощью ключевых слов, список - это ключевое слово.Может возникнуть проблема, и вы даже сможете найти причину.

0 голосов
/ 25 декабря 2018

Возможно использовать zip:

for a,b,c in zip(scores,boxes,class_list):
   if a >= 0.98:
       data[k+1] = {"score":a,"box":b, "class": c } ;
       k=k+1;
print("Final_result",data)

Вывод:

Final_result {2: {'score': 0.98, 'box': [0.1, 0.2, 0.3, 0.4], 'class': 1}}

Редактировать:

for a,b,c in zip(scores,boxes,class_list):
   if a >= 0.98:
       data[k+1] = {"score":a,"box":b, "class": int(c) } ;
       k=k+1;
print("Final_result",data)
...