从python3.4中的列表中返回所有对

2024-04-26 09:57:22 发布

您现在位置:Python中文网/ 问答频道 /正文

这段代码在Python2.7中可以工作,但在3.4中不行。在

import numpy as np
import itertools

a = [[3,7,9],[2,7,5],[6,9,5]]

def all_pairs(a, top=None):
    d = collections.Counter()
    for sub in a:
        if len(a)<2:
            continue
        sub.sort()
        for comb in itertools.combinations(sub,2):
            d[comb]+=1
    return sorted(np.array([i[0] for i in d.most_common(top)]).tolist())

这是我期望的结果:

^{pr2}$

但是使用Python 3.4,我得到了以下回溯:

^{3}$

另外,当我只添加一个元素时,我什么也得不到:

all_pairs([3,7,9])
[]

# Expected to get:
[[3,7],[3,9],[7,9]]

有没有更好的方法来编写这段代码来解决这两个问题?在


Tags: 代码inimportnumpynonefortopdef
1条回答
网友
1楼 · 发布于 2024-04-26 09:57:22

在python2中,map()生成一个list,它的方法是sort()。在python3中,map()生成一个map对象,它充当一个惰性生成器,没有{}方法。如果要在其上使用sort(),必须先将该对象传递给list()

sub = list(sub)
sub.sort()

当然,如果您这样做的话,您还可以使用sorted(),它可以处理python3map对象和list对象:

^{pr2}$

相关问题 更多 >