python - 如何将 Pandas 数据框列连接到可迭代列表中？

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)

应该可以。