4

我正在尝试编写一个算法：

我有如下OrderedDict数据类型的输入：

odict_items([(3, [(0, 1), (1, 1), (1, 1)]), (11, [(0, 0), (1, 1), (1, 1)]), (12, [(0, 0), (1, 1), (1, 1)])])

我正在尝试编写一个函数来在每个中添加相同元组的数量，key例如预期的输出，如下所示：如果有(1,1)相同的元组1，那么如果有两次2，那么一个：

odict_items([(3, [(0, 1), (1, 1), (1, 1)],2), (11, [(0, 0), (1, 1), (1, 1)],2), (12, [(0, 0), (1, 0), (1, 1)]),1])

这是我的尝试，但如何改进它并将其添加到OrderedDict？

def foo(OrderedDict):
    listOfDic = list(makeDataStruc().items())
    tupleCounter = 0
    for i in range(len(listOfDic[1])):
        if listOfDic[1][1][i][0] == 1 and listOfDic[1][1][i][1] == 1:
            tupleCounter += 1
    return tupleCounter

我在哪里犯错？

4

1 回答 1

0

我做出以下假设，

  • 您只是想将计数添加(1,1)到OrderedDict值中
  • 您不想创建一个新的数据结构来覆盖当前的OrderedDict.
  • 可以修改原来的字典

现在基于问题中可用的信息和上述假设，一种可能的解决方案是用包含两个元素的列表替换每个值，即`[原始值，(1, 1) 的计数]

from collections import OrderedDict

odict_items = [(3, [(0, 1), (1, 1), (1, 1)]),
               (11, [(0, 0), (1, 1), (1, 1)]),
               (12, [(0, 0), (1, 1), (1, 1)])]

my_odict = OrderedDict()

for item in odict_items:
    k, v = item
    my_odict[k] = v

现在计算(1,1)每个值中的出现并相应地更新值

pattern_to_find = (1, 1)

for key, value in my_odict.items():
    tuple_count = value.count(pattern_to_find)
    new_value = [value, tuple_count]
    my_odict[key] = new_value

现在字典有以下内容：

OrderedDict([(3, [[(0, 1), (1, 1), (1, 1)], 2]),
             (11, [[(0, 0), (1, 1), (1, 1)], 2]),
             (12, [[(0, 0), (1, 1), (1, 1)], 2])])

现在您可以创建附加函数来仅访问值或元组计数

# Returns the count of (1, 1) only
def get_count(my_dict, key):
    return my_dict[key][1]

# Return the original value only
def get_tuple(my_dict, key):
    return my_dict[key][0]

所以你可以像这样使用它们

print(my_odict[3])
# [[(0, 1), (1, 1), (1, 1)], 2]

print(get_count(my_odict,3))
# 2

print(get_tuple(my_odict, 3))
# [(0, 1), (1, 1), (1, 1)]
于 2020-02-18T15:40:05.133 回答