如何将pandas数据帧列连接到可迭代列表？

2019年7月21日 105次阅读

我正在使用大约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

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

最佳答案 你似乎有一个错误的价值楠,我假设是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)

应该管用.

点赞