Я хочу классифицировать некоторые данные в соответствии с моими правилами, которые не полностью соответствуют моим правилам - PullRequest
0 голосов
/ 07 апреля 2019

Я хочу классифицировать некоторые данные в соответствии с моими правилами, которые не полностью соответствуют моим правилам.

group = ((1, ["zt"]), (2, ["xjt"]), (3, ["ppt"]), (4, ["dpt"]), (5, ["sxb"]), (6, ["tmzt"]))

indata = [(1001,"zt_1"), (1002,"zt_2"), (1021,"tmzt_1"),(1023,"xjt_1"), (1244,"xjt_2")]

Я хочу группировать по правилам (группа)

outdata = [(1, 1001,"zt_1", 1), (1, 1002,"zt_2", 2), (6, 1021,"tmzt_1", 1),(2, 1023,"xjt_1", 1), (2, 1244,"xjt_2", 2)]

outdataсодержать (идентификатор группы, идентификатор индата, имя индата, внутренняя сортировка группы), но не все данные соответствуют правилам моей группы.

выходные данные также могут иметь формат, подобный

[{"group_id": 1, "image_id": 1001, "name": "zt_1", "sort_order": 1}, ...] 

группа уникальна

1 Ответ

1 голос
/ 07 апреля 2019

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

group = ((1, ["zt"]), (2, ["xjt"]), (3, ["ppt"]), (4, ["dpt"]), (5, ["sxb"]), (6, ["tmzt"]))

indata = [(1001,"zt_1"), (1002,"zt_2"), (1021,"tmzt_1"),(1023,"xjt_1"), (1244,"xjt_2")]

group_dict = {tup[1][0]: tup[0] for tup in group}
#Output:
{'zt': 1, 'xjt': 2, 'ppt': 3, 'dpt': 4, 'sxb': 5, 'tmzt': 6}

result = []
for tup in indata:
    key, val = tup[1].split("_")
    if key in group_dict:
        result.append((group_dict[key], *tup, int(val)))

print(result)
#Output:
[(1, 1001, 'zt_1', 1),
 (1, 1002, 'zt_2', 2),
 (6, 1021, 'tmzt_1', 1),
 (2, 1023, 'xjt_1', 1),
 (2, 1244, 'xjt_2', 2)]
...