Python:将pd.Series的所有组合写入文本文件中

时间:2016-08-07 17:18:31

标签: python pandas combinations

我有几个Pandas系列的独特字符串:

    First Series
    P0A8V2
    P36683
    P15254

    Second Series
    P09831
    P0AFG8

我想编写一个看起来像这样的文本文件(tab seperator):

P0A8V2 P36683
P0A8V2 P15254
P36683 P15254 
P09831 P0AFG8

因此,在一个系列中,每个对象只相互组合一次。顺序无关紧要。然后下一个系列马上开始。

有一种简单的方法吗?

更新:

字符串位于DataFrame的索引中。我用df.index.values访问它们。 DataFrame位于df_list。

def Cluster_Network(df_list):
    combi_list = []
    for cluster in df_list:
       combi_list.append(tuple(itertools.combinations(cluster.index.values, 2)))
    return combi_list

我得到了一对元组列表。

  [('P77717', 'P10378'),
  ('P18393', 'P77444'),
  ('P18393', 'P0AD44'),
  ('P18393', 'P10378'),
  ('P77444', 'P0AD44'),
  ('P77444', 'P10378'),
  ('P0AD44', 'P10378')),
 (('P77562', 'P41039'),)]

如何从该列表中编写文本文件?

2 个答案:

答案 0 :(得分:3)

看起来你快到了。

combi_list = []
for cluster in df_list:
    combi_list.append(pd.DataFrame(list(itertools.combinations(cluster.index, 2))))
result_df = pd.concat(combi_list, ignore_index=True)
result_df.to_csv(filename, sep='\t', index=False, header=False) 

这将生成如下文件:

P0A8V2  P36683
P0A8V2  P15254
P36683  P15254
P09831  P0AFG8

答案 1 :(得分:1)

另一种方法是使用转换系列列表,然后使用itertools.combinations获得所需的结果......就像这样,

import pandas as pd
s1  = pd.Series(['a', 'b', 'c'])
s2  = pd.Series(['d', 'e'])

import itertools
s= s1.tolist()
s.extend(s2.tolist())
open('test.txt','w').writelines(["%s\t%s\n" % (item[0], item[1])  for item in list(itertools.combinations(s,2))])