从多个列表读取Pandas数据帧

Question

我有一个包含如下数据的输入文件：

[ABCD,color,NORMAL,N,2015-02-20,1]
[XYZA,color,NORMAL,N,2015-05-04,1]
[GFFD,color,NORMAL,N,2015-07-03,1]
[NAAS,color,NORMAL,N,2015-08-26,1]
[LOWW,color,NORMAL,N,2015-09-26,1]
[KARA,color,NORMAL,N,2015-11-08,1]
[ALEQ,color,NORMAL,N,2015-12-04,1]
[VDDE,color,NORMAL,N,2015-12-23,1]
[QWER,color,NORMAL,N,2016-01-18,1]
[KDSS,color,NORMAL,Y,2015-08-29,1]

我已经创建了一个变量来存储列名，比如table\u columns。如何将其转换为熊猫数据帧？在

应该非常类似于从多个列表中读取，但是它将从中读取的列表数量是n

Answer 1

您可以用^{}和^{}来尝试^{}：

import pandas as pd
import io

temp=u"""[ABCD,color,NORMAL,N,2015-02-20,1]
[XYZA,color,NORMAL,N,2015-05-04,1]
[GFFD,color,NORMAL,N,2015-07-03,1]
[NAAS,color,NORMAL,N,2015-08-26,1]
[LOWW,color,NORMAL,N,2015-09-26,1]
[KARA,color,NORMAL,N,2015-11-08,1]
[ALEQ,color,NORMAL,N,2015-12-04,1]
[VDDE,color,NORMAL,N,2015-12-23,1]
[QWER,color,NORMAL,N,2016-01-18,1]
[KDSS,color,NORMAL,Y,2015-08-29,1]"""
#after testing replace io.StringIO(temp) to filename
df = pd.read_csv(io.StringIO(temp),header=None,names=['a','b','c','d','e','f'])

#remove []
df['a'] = df['a'].str.strip('[')
df['f'] = df['f'].str.strip(']')
#convert column e to datetime
df['e'] = pd.to_datetime(df['e'])
print df
      a      b       c  d          e  f
0  ABCD  color  NORMAL  N 2015-02-20  1
1  XYZA  color  NORMAL  N 2015-05-04  1
2  GFFD  color  NORMAL  N 2015-07-03  1
3  NAAS  color  NORMAL  N 2015-08-26  1
4  LOWW  color  NORMAL  N 2015-09-26  1
5  KARA  color  NORMAL  N 2015-11-08  1
6  ALEQ  color  NORMAL  N 2015-12-04  1
7  VDDE  color  NORMAL  N 2015-12-23  1
8  QWER  color  NORMAL  N 2016-01-18  1
9  KDSS  color  NORMAL  Y 2015-08-29  1

Answer 2

一种方法是：

from StringIO import StringIO

import pandas as pd

>>> pd.read_csv(StringIO(open('stuff.txt').read().replace('[', '').replace(']', '')), header=None)
    D   color   NORMAL  N   2015-02-20  1
0   XYZA    color   NORMAL  N   2015-05-04  1
1   GFFD    color   NORMAL  N   2015-07-03  1
2   NAAS    color   NORMAL  N   2015-08-26  1
3   LOWW    color   NORMAL  N   2015-09-26  1
4   KARA    color   NORMAL  N   2015-11-08  1
5   ALEQ    color   NORMAL  N   2015-12-04  1
6   VDDE    color   NORMAL  N   2015-12-23  1
7   QWER    color   NORMAL  N   2016-01-18  1
8   KDSS    color   NORMAL  Y   2015-08-29  1