如何将pandas dataframe列连接到可迭代列表？

Question

我正在使用大约200列的pandas数据框。第31-73列看起来像这样：

31   32   33
NaN  NaN  Z
X    NaN  Z
NaN  NaN  NaN
NaN  Y    Z

我希望返回的是一个看起来像这样的新专栏：

new_col
[Z]
[X,Z]
[]
[Y,Z]

我很近，但并不完全在那里。这段代码：

data['new_col'] = data.ix[:, 30:73].fillna('').apply(lambda row: ','.join(map(str, row)), axis=1)

返回：

new_col
,,Z
,X,,Z
,,,
,,Y,Z

如何从新的连接列表中排除不必要的分隔符？

Answer 1

您似乎有一个错误的值Nan，我假设它是NaN，无论如何这应该有效：

In [24]:
df.apply(lambda x: [','.join(x.dropna())], axis=1)

Out[24]:
0      [Z]
1    [X,Z]
2       []
3    [Y,Z]
dtype: object

所以在你的情况下：

data['new_col'] = data.apply(lambda x: [','.join(x.dropna())], axis=1)

应该有用。