3

我试图迭代一个数组两次以获得成对的元素（e_i，e_j），但我只想要 i < j 的元素。

基本上，我想要的是类似 C 的语言。

int my_array[N] = ...;
for (int i=0; i<N; i++)
        for (int j=i+1; j<N; j++)
                something(my_array[i],my_array[j]);

我没有在itertools中找到我想要的东西（我找到的最接近的东西是itertools.product(*iterables[, repeat])）。

我尝试了一些事情，但我并不真正相信其中任何一个：

my_list=range(10)

# Using enumerate and slices - explicit loop
res=[]
for i,j in enumerate(my_list):
        for k in my_list[i+1:]:
                res.append((j,k))
print res

# Using enumerate and slices - list comprehension
res=[(j,k) for i,j in enumerate(my_list) for k in my_list[i+1:]]
print res

# Using enumerate and xrange - explicit loop
res=[]
for i,j in enumerate(my_list):
        for k in range(i+1, len(my_list)):
                res.append((j,my_list[k]))
print res       

# Using enumerate and xrange - list comprehension
res=[(j,my_list[k]) for i,j in enumerate(my_list) for k in range(i+1, len(my_list))]
print res

我仍然相信有一个更好、更 Pythonic 的解决方案。欢迎任何建议。

4

3 回答 3

9

只需使用itertools.combinations(my_list, 2).

于 2013-07-21T02:13:47.230 回答
2

您不能只使用该range功能并使用：

vect = [...]
for i in range(0, len(vect)):
    for j in range(i+1, len(vect)):
        do_something()
于 2013-07-21T02:00:41.873 回答
0

我建议如下：

midIdx = len(mylist) / 2
[ dosomehing(ele_i, ele_j) for ele_i, ele_j in 
    zip( mylist[0:midIdx], mylist[midIdx + 1:len(mylist)] ) ]

对于大多数解释型语言，for 循环并不是最佳选择。Python 提供了列表推导，更具可读性和效率。

于 2013-07-21T02:17:28.007 回答