Python pandas将逗号分隔值列表转换为dataframe

时间:2016-11-07 15:46:23

标签: python pandas dataframe

我有一个字符串列表,如下所示:

["Name: Alice, Department: HR, Salary: 60000", "Name: Bob, Department: Engineering, Salary: 45000"]

我想将此列表转换为如下所示的DataFrame:

Name | Department | Salary
--------------------------
Alice | HR | 60000

Bob | Engineering | 45000

最简单的方法是什么? 我的直觉说将数据放入CSV并用正则表达式“^。*:”分隔标题,但必须有一个更简单的方法

3 个答案:

答案 0 :(得分:8)

通过一些字符串处理,您可以获得一个dicts列表并将其传递给DataFrame构造函数:

lst = ["Name: Alice, Department: HR, Salary: 60000", 
       "Name: Bob, Department: Engineering, Salary: 45000"]
pd.DataFrame([dict([kv.split(': ') for kv in record.split(', ')]) for record in lst])
Out: 
    Department   Name Salary
0           HR  Alice  60000
1  Engineering    Bob  45000

答案 1 :(得分:3)

你可以这样做:

In [271]: s
Out[271]:
['Name: Alice, Department: HR, Salary: 60000',
 'Name: Bob, Department: Engineering, Salary: 45000']

In [272]: pd.read_csv(io.StringIO(re.sub(r'\s*(Name|Department|Salary):\s*', r'', '~'.join(s))),
     ...:             names=['Name','Department','Salary'],
     ...:             header=None,
     ...:             lineterminator=r'~'
     ...: )
     ...:
Out[272]:
    Name   Department  Salary
0  Alice           HR   60000
1    Bob  Engineering   45000

答案 2 :(得分:3)

有点创意

s.str.extractall(r'(?P<key>[^,]+)\s*:(?P<value>[^,]+)') \
    .reset_index('match', drop=True) \
    .set_index('key', append=True).value.unstack()

enter image description here

设置

l = ["Name: Alice, Department: HR, Salary: 60000",
     "Name: Bob, Department: Engineering, Salary: 45000"]
s = pd.Series(l)