当某些列列表值为空时,如何合并包含列表值的列?

时间:2016-11-08 19:59:27

标签: python list pandas array-merge

我想以这种方式合并数据框中的列:

list_A      list_B       
A, T, G     G, C        
            B, K         
C, L, AG    L, AG, K     
F, K                 

输出应为:

list_A      list_B       list_A&B
A, T, G     G, C         A, T, G, C
            B, K         B, K
C, L, AG    L, AG, K     C, L, AG, K
F, K                     F, K

我做了:

df['list_A&B'] = df['list_A'].astype(list) + ', ' + df['list_B'].astype(list)

而且,我得到了:

     list_A    list_B            list_A&B
0   A, T, G      G, C       A, T, G, G, C
1       NaN      B, K                 NaN
2  C, L, AG  L, AG, K  C, L, AG, L, AG, K
3      F, K       NaN                 NaN

当其中一个列表为空时,列表的并集是一个问题。但是,为什么?

然后我尝试将其作为字符串进行联合,但现在添加的nan不能被删除: df['list_A&B'] = df['list_A'].astype(str) + ', ' + df['list_B'].astype(str)

给出:

     list_A    list_B            list_A&B
0   A, T, G      G, C       A, T, G, G, C
1       NaN      B, K           nan, B, K
2  C, L, AG  L, AG, K  C, L, AG, L, AG, K
3      F, K       NaN           F, K, nan

有了这个输出,我很难删除' nan'因为它们被报告为字符串而dropna()和fillna()不能使用它。

任何建议! - K

1 个答案:

答案 0 :(得分:1)

通过我对您目前提供的数据的解释,看起来这些只是字符串值。

<强> 步骤:

使用str.catsep=','na_rep=''逐行连接它们,它将散布在其他字符串字符中的NaN值视为空字符

然后,删除其中不需要的空格,并在使用filter(None,...)获取列表的唯一元素后使用set消除空字符串。

最后,加入结果列表,使其成为列表的字符串表示形式,但不包括括号。

df['list_A&B'] = df['list_A'].str.cat(df['list_B'], ',','') \
                  .apply(lambda x: ', '.join(list(filter(None, set(re.sub(r"\s+", "", x) \
                  .split(','))))))

df                  # df.fillna('') to get back your desired output

enter image description here

开始使用DF

df = pd.DataFrame({'list_A': ['A, T, G', np.NaN, 'C, L, AG ', 'F, K'], 
                   'list_B': ['G, C', 'B, K', 'L, AG, K', np.NaN]})
df

enter image description here

相关问题