所有从numpy数组中设置的长度为2

Question

这看起来像是一个罐头的numpy或scipy函数......但我无法找到它。

我有一个元组数组，其中元组包含类的均值和标准差。我需要第二个数组与所有独立的元组相互组合（所以所有的数组长度为2的子集）。

所以，例如：

original = [(0.5,0.112),(2.3,0.1),(5,0.7)] 

我需要：

subsets = [((0.5,0.112),(2.3,0.1)),/ 
      ((0.5,0.112),(5,0.7)),/ 
      ((2.3,0.1),(5,0.7))] 

为任意lengthed原始阵列。

我有什么现在：

def subsets_length_2(vector): 
    subset_vector = [] 
    for i in vector: 
     for j in vector: 
      if i != j: 
       subset_vector.append((i,j)) 
    subset_vector = np.asarray(np.unique(subset_vector)) 
    return subset_vector 

Answer 1

您可以使用itertools.combinations：

import itertools 
subsets = list(itertools.combinations(original, 2)) 
print subsets 
# [((0.5, 0.112), (2.3, 0.1)), ((0.5, 0.112), (5, 0.7)), ((2.3, 0.1), (5, 0.7))] 

Answer 2

您仍然可以使用双重for循环，但它在没有必要循环整个数组与内部for循环。

def subsets_length_2(vector): 
    subset_vector = [] 
    n = len(vector) 
    for i in range(n-1): 
     for j in range(i+1, n): 
      subset_vector.append((vector[i],vector[j])) 
    return np.asarray(subset_vector)