如何将一列数据框架中的嵌套列表转换为一维列表?

时间:2018-01-18 20:25:55

标签: python list pandas dataframe

   videoID  long   lat   viewerCount
0      225   -10   1.8     [1,[3,4]]
1      228    12  23.0         [5,5]
2      123    10  20.0  [1,[2, [3]]]

我有如上所示的数据帧。我要做的是将viewerCount的值转换为单个Dimension数组。预期输出如下所示:

   videoID  long   lat viewerCount
0      225   -10   1.8   [1, 3, 4]
1      228    12  23.0      [5, 5]
2      123    10  20.0   [1, 2, 3]

我尝试过很多资源但未能解决问题。请帮帮我。在此之前,要感谢能解决它的生命保护者。

1 个答案:

答案 0 :(得分:1)

将其转换为字符串,删除所有方括号和空格,最后将其拆分为','

In [28]: df['viewerCount'] = \
             df['viewerCount'].astype(str).str.replace(r'[\[\s\]]', '').str.split(',')

In [29]: df
Out[29]:
   videoID  long   lat viewerCount
0      225   -10   1.8   [1, 3, 4]
1      228    12  23.0      [5, 5]
2      123    10  20.0   [1, 2, 3]