我想以这种方式合并数据框中的列:
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
答案 0 :(得分:1)
通过我对您目前提供的数据的解释,看起来这些只是字符串值。
<强> 步骤: 强>
使用str.cat
与sep=','
和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
开始使用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