2016-04-11 78 views
2

我有一个pandas.core.frame.DataFrame,看起来像这样:大熊猫据帧爆炸列内容

  0 1 
0 [1,2,3] 1 
1 [2,2,1] 1 
2 [1,2,1] 1 
... 

的最后一列是标签,每个柱下“0”的阵列应该是给定类不同的数据点。

我希望这可以变成:

x0 x1 x2 label 
0 1 2 3 1 
1 2 2 1 1 
2 1 2 1 1 

我都试过,没有运气

ds = ds.apply(lambda x: numpy.ravel(x)) 

这是下面的结果如下,显然是不应该做的方式这个。

<list>.extend(zip(points,labels)) 
ds = pandas.core.frame.DataFrame(data=<list>) 

有关如何修复实际数据集或正确创建具有两个列表(点和标签)的任何帮助,请参阅。

回答

2

这是我会怎么做。首先删除您的1列(所以我们不乱的命名):

df['id'] = df[1] 
df = df.drop(1, axis = 1) 

然后创建一个OBJ文件,与我们要Concat的东西,CONCAT:

objs = [df, pd.DataFrame(df[0].tolist())] 
pd.concat(objs, axis=1) 



      0 id 0 1 2 
0 [1, 2, 3] 1 1 2 3 
1 [2, 2, 1] 1 2 2 1 
2 [1, 2, 1] 1 1 2 1 
+0

此解决方案效果最好!我修改了第二部分:'pd.concat([df,pd.DataFrame(list(df [0]))],axis = 1)',似乎比使用'.tolist( )'。谢谢! – Thanos

0

您可以以不同方式创建数据框以获得所需内容而不是试图展开列。请参见下面的代码,

import pandas as pd 
points = [[1,2,3],[2,2,1],[1,2,1]] 
labels = [1,1,1] 
x0 = [p[0] for p in points] 
x1 = [p[1] for p in points] 
x2 = [p[2] for p in points] 
df = pd.DataFrame({'x0': x0,'x1': x1,'x2': x2, 'label': labels}) 
print (df) 

搞定,

label x0 x1 x2 
0  1 1 2 3 
1  1 2 2 1 
2  1 1 2 1 
-1

最好的,我可以提供:

import numpy as np 
# first convert your lists to an array, then iterate 
tmp = np.array(df[0].tolist()) 

for r in np.arange(0,3): 
    df['x' + str(r)] = tmp[:,r] 
1

我假设你的当前列标题是文本而不是整数。

df2 = pd.concat([pd.DataFrame(zip(*df['0'])), df['1']], axis=1) 
df2.columns = ['x' + str(c) for c in df2.columns[:-1]] + ['label'] 

>>> df2 
    x0 x1 x2 label 
0 1 2 1  1 
1 2 2 2  1 
2 3 1 1  1 

*运营商使用zip解压名单:

>>> zip(*df['0']) 
[(1, 2, 1), (2, 2, 2), (3, 1, 1)] 

所以,你可以轻松扩展数据帧:

>>> pd.DataFrame(zip(*df['0'])) 
    0 1 2 
0 1 2 1 
1 2 2 2 
2 3 1 1 

,那么你只需要连接的最后一列,并重新命名所有的列。

+0

列是整数。当我尝试pd.DataFrame(zip(* ds [0]))时,我得到以下错误:frame.py 283 mgr = self._init_dict({},索引,列,dtype = dtype) 284 elif isinstance(data, (数据参数不能为迭代器) 286 else: 287 try: TypeError:数据参数不能是迭代器。有任何想法吗? – Thanos