迭代python集中的各个元素

时间:2016-01-15 10:49:12

标签: python python-2.7 set iteration hashtable

给定m 设置具有n个元素的整数。

我有以下代码,它输出最多次出现的元素。

def find_element_which_appeared_in_max_sets(input_set):

    hash_table = {}

    # count the frequencies
    for pair in input_set:
        for i in range(0,len(pair)):
            if pair[i] in hash_table:
                hash_table[pair[i]] = hash_table[pair[i]] + 1
            else:
                hash_table[pair[i]] = 1 # first occurence

    # scan and find the element with highest frequency.
    max_freq = 0
    for elem in hash_table:

        if hash_table[elem] > max_freq:
            max_freq = hash_table[elem]
            max_occured_elem = elem

    return max_occured_elem


input_set = {(5,4),(3,2),(4,3),(8,3)}
print ""+str(find_element_which_appeared_in_max_sets(input_set))

输出:

3

是否有更整洁/优雅的方式迭代整个集合中的各个元素?

2 个答案:

答案 0 :(得分:4)

您可以简单地使用collections.Counteritertools.chain.from_iterable,就像这样

def find_element_which_appeared_in_max_sets(input_set):
    return Counter(chain.from_iterable(input_set)).most_common(1)[0][0]

chain.from_iterable(input_set)将使元组的输入集变得扁平,以获得单个迭代,从而逐个给出每个元组的值。

然后Counter计算每个项目发生的次数,并将项目及其计数维护为字典。

然后most_common(1)上的Counter调用会返回一个项目列表,其中包含最高出现次数n(参数传递给它)的最大出现次数,格式为(item, count)。由于我们只对该项感兴趣,因此我们会使用[0][0]返回第一项。

答案 1 :(得分:4)

仅使用内置函数,不使用标准库导入:

def find_element_which_appeared_in_max_sets(input_set):
    hash_table = {}
    for pair in input_set:
        for item in pair:
            #enhanced as proposed by thefourtheye
            hash_table[item] = hash_table.get(item, 0) + 1
    max_occured_element = max(hash_table, key=hash_table.get)
    return max_occured_element