1

我在 Python 中有以下列表：

l = [[2], [3], [2, 2], [5], [2], [3], [7], [2, 2, 2], [3, 3], [2], [5], [11], [2, 2], [3], [13], [2], [7], [3], [5], [2, 2, 2, 2], [17], [2], [3, 3], [19], [2, 2], [5]]

我想编写一个函数来返回最大长度的唯一值子列表。在这种情况下，函数将返回：

l = [[5], [7], [3, 3], [11], [13], [2, 2, 2, 2], [17], [19]]

我仍然是 python 的初学者，但是我对如何编写这样的函数知之甚少。我得到的最远的是弄清楚我可以通过使用嵌套循环来迭代子列表。但是从我对 Python 的了解来看，似乎必须有一些比使用循环更简单的方法来返回我正在寻找的列表。

更新：

这是我对代码所做的事情：解决项目 euler #5，非蛮力方式！

我确信这段代码可以重构，但无论如何。

谢谢你们的帮助，伙计们。itemgetter正是我所需要的。

#!/usr/bin/python
# coding = UTF-8

import argparse, sys, math
from itertools import groupby
from collections import defaultdict
from operator import itemgetter

parser = argparse.ArgumentParser()
parser.add_argument('filename', nargs='?')
args = parser.parse_args()
if args:
   intinput = int(sys.argv[1])
elif not sys.stdin.isatty():
    intinput = int(sys.stdin.read())
else:
    parser.print_help()

def prime_factorize(n):
    factors = []
    number = math.fabs(n)

    while number > 1:
        factor = get_next_prime_factor(number)
        factors.append(factor)
        number /= factor

    if n < -1: 
        factors[0] = -factors[0]

    return factors

def get_next_prime_factor(n):
    if n % 2 == 0:
        return 2

    for x in range(3, int(math.ceil(math.sqrt(n)) + 1), 2):
        if n % x == 0:
            return x

    return int(n)


def mkfactors(n):
  tpf = []
  for i in range(n+1):
    tpf.extend(prime_factorize(i))
  return tpf

l = [list(g) for k,g in groupby(mkfactors(intinput))]

m = [max(g) for _,g in groupby(sorted(l,key=itemgetter(0)),key=itemgetter(0))]


prod = 1

for list in m:
  for element in list:
    prod *= element

print prod
4

6 回答 6

2

如果你知道每个子列表都有相同的元素，你可以这样做：

l = [[2], [3], [2, 2], [5], [2], [3], [7], [2, 2, 2], [3, 3], [2], [5], [11], [2, 2], [3], [13], [2], [7], [3], [5], [2, 2, 2, 2], [17], [2], [3, 3], [19], [2, 2], [5]]

from collections import defaultdict

my_dict = defaultdict(list)

for ele in l:
    if len(my_dict[ele[0]]) < len(ele):
        my_dict[ele[0]] = ele

结果：

>>> my_dict.values()
[[2, 2, 2, 2], [3, 3], [5], [7], [11], [13], [17], [19]]
于 2013-01-11T00:23:20.577 回答
2

这里最简单的做法是使用使问题变得简单的数据结构，然后您总是可以在事后转换回来。

例如，dict将键（质数）映射到长度（指数）很容易。所以：

>>> l = [[2], [3], [2, 2], [5], [2], [3], [7], [2, 2, 2], [3, 3], [2], [5], [11], [2, 2], [3], [13], [2], [7], [3], [5], [2, 2, 2, 2], [17], [2], [3, 3], [19], [2, 2], [5]]
>>> d = {}
>>> for sublist in l:
...     value, count = sublist[0], len(sublist)
...     if count > d.get(value, 0):
...         d[value] = count
>>> d
{2: 4, 3: 2, 5: 1, 7: 1, 11: 1, 13: 1, 17: 1, 19: 1}

如何将它转回 a listof lists 应该很明显，所以我将把它留给你。

请注意，这会丢失订单，但您可以使用OrderedDict. 它也失去了list身份——例如，[2, 2, 2, 2]你最后得到的 将等于，但与原来的 不同[2, 2, 2, 2]。但这也很容易解决——只需sublist直接存储而不是使用count. 无论如何，我认为这些都与您的问题无关。

于 2013-01-11T00:23:37.267 回答
1 
l = [[2], [3], [2, 2], [5], [2], [3], [7], [2, 2, 2], [3, 3], [2], [5], [11], [2, 2], [3], [13], [2], [7], [3], [5], [2, 2, 2, 2], [17], [2], [3, 3], [19], [2, 2], [5]]
l = [max(i for i in l if j in i) for j in (2, 3, 5, 7, 11, 13, 17, 19)]
print(l)
# [[2, 2, 2, 2], [3, 3], [5], [7], [11], [13], [17], [19]]

我想这只是一个带有嵌套 for 循环的列表理解，但它工作正常。

于 2013-01-11T00:27:16.677 回答
1

使用collections.Counter和sets：

In [47]: s=set([x[0] for x in lis])

In [48]: c=[Counter(x) for x in lis]

In [49]: [max(c,key=lambda y:y[x]) for x in s]
Out[49]: 
[Counter({2: 4}),
 Counter({3: 2}),
 Counter({5: 1}),
 Counter({7: 1}),
 Counter({11: 1}),
 Counter({13: 1}),
 Counter({17: 1}),
 Counter({19: 1})]

其他方式：

In [64]: from collections import defaultdict

In [65]: d=defaultdict(list)

In [66]: for x in lis:
    d[x[0]].append(len(x))
   ....:     

In [67]: [[x]*max(y) for x,y in d.items()]
Out[67]: [[2, 2, 2, 2], [3, 3], [5], [7], [11], [13], [17], [19]]
于 2013-01-11T00:36:49.267 回答
1 
from itertools import groupby
from operator import itemgetter

[max(g) for _,g in groupby(sorted(l),key=itemgetter(0))]

出去：

[[2, 2, 2, 2], [3, 3], [5], [7], [11], [13], [17], [19]]
于 2013-01-11T01:02:28.913 回答
1

一个直接的解决方案是将排序列表转换为以键作为列表的第一个元素的字典，最终将根据键删除重复项。

>>> {e[0]: e for e in sorted(l)}.values()
[[2, 2, 2, 2], [3, 3], [5], [7], [11], [13], [17], [19]]

对于 Python 版本 < 2.7，其中 dict 理解不可用

>>> dict((e[0], e) for e in sorted(l)).values()
[[2, 2, 2, 2], [3, 3], [5], [7], [11], [13], [17], [19]]
于 2013-01-11T04:26:35.093 回答