list - 如何从 Pandas 数据框中制作列表列表，跳过 nan 值

Question

我有一个看起来大致像的 Pandas 数据框

    foo   foo2   foo3  foo4
a   NY    WA     AZ    NaN
b   DC    NaN    NaN   NaN
c   MA    CA     NaN   NaN

我想对这个数据框的观察进行嵌套列表，但省略 NaN 值，所以我有类似 [['NY','WA','AZ'],['DC'],[' MA',CA']。

此数据框中有一个模式，如果这有所不同，则如果 fooX 为空，则后续列 fooY 也将为空。

我最初有类似下面的代码。我确定有更好的方法来做到这一点

A = [[i] for i in subset_label['label'].tolist()]
B = [i for i in subset_label['label2'].tolist()]
C = [i for i in subset_label['label3'].tolist()]
D = [i for i in subset_label['label4'].tolist()]
out_list = []
for index, row in subset_label.iterrows():
out_list.append([row.label, row.label2, row.label3, row.label4])
out_list

Answer 1

试试这个:

In [77]: df.T.apply(lambda x: x.dropna().tolist()).tolist()
Out[77]: [['NY', 'WA', 'AZ'], ['DC'], ['MA', 'CA']]