将列表转换为1列熊猫数据帧

时间:2015-08-21 10:36:23

标签: python pandas

我有一个包含多行的文件。我正在读每行,分割每个单词/数字并存储在列表中。在此之后,我试图将此列表转换为1列熊猫数据帧。

然而,在运行我的代码后,我只得到一行满是列表。我需要的是 1列,行数可变,有一些值。

以下是我写的代码段:

for line1 in file:
    test_set=[]

    test_set.append(next(file).split())

    df1 = DataFrame({'test_set': [test_set]})

我的输出类似于:

              test_set
0  [[1, 0, 0, 0, 0, 0, 1, 1, 1, 0]]

但我想要的是

    test_set
0 1
1 0
2 0
3 0
4 0
5 0
6 1
7 1
8 1
9 0

有什么建议我做错了或者我该如何实现?感谢。

输入数据样本摘录

id1 id2 id3 id4
0 1 0 1 
1 1 0 0

id10 id5 id6 id7
1 1 0 1 
1 0 0 1

.
.
.

3 个答案:

答案 0 :(得分:13)

你想要这个:

df1 = DataFrame({'test_set': test_set})

无需再次将列表包装在另一个列表中,通过这样做,您可以有效地说明您的df数据是一个包含单个元素的列表,这是另一个列表。

修改

查看输入数据,您只需加载它,然后将df构建为单列,如下所示:

In [134]:
# load the data
import io
import pandas as pd
t="""id1 id2 id3 id4
0 1 0 1 
1 1 0 0"""
df = pd.read_csv(io.StringIO(t), sep='\s+')
df

Out[134]:
   id1  id2  id3  id4
0    0    1    0    1
1    1    1    0    0

现在转置df并执行列表理解,这将构建您的列表并使用pd.concat连接它们:

In [142]:    
pd.concat([df.T[x] for x in df.T], ignore_index=True)

Out[142]:
0    0
1    1
2    0
3    1
4    1
5    1
6    0
7    0
dtype: int64

答案 1 :(得分:4)

这应该没问题:

df1 = DataFrame({'test_set': test_set})

test_set已经是一个列表,你不必遍历它,所以你可以把它作为一个值添加到pandas中。

print df1

test_set
0 1
1 0
2 0
3 0
4 0
5 0
6 1
7 1
8 1
9 0

答案 2 :(得分:1)

结果我只需添加此

df1 = DataFrame({'test_set': value for value in test_set})

但是我仍然希望得到一个成本较低的答案,因为这也会增加另一个因素的复杂性,或者'n'不够好。