如何在熊猫中有条件地从不同的列中进行选择

Question

我有一个像Nx5形状的熊猫DataFrame

['','','A','','']
['','C','','','']
['','A','','','']
['','','','T','']
.
.
.

我想将其转换为Nx1形状以获得非空值

['A']
['C']
['A']
['T']
.
.
.

那怎么办？

Answer 1

您可以为每一行调用"".join ：

df.apply("".join, axis=1)

如果不确定每一行只有一个非空值，则以下方法更好：

import pandas as pd

df = pd.DataFrame(
    [['','','A','',''],
    ['','C','','',''],
    ['','A','','',''],
    ['','','','T','']]
)

s = df.stack()
print s[s!=""]

输出：

0  2    A
1  1    C
2  1    A
3  3    T
dtype: object

超过一栏：

r = s[s!=""]
r.groupby(level=0).apply(pd.DataFrame.reset_index, drop=True).unstack()