如何使用pandas读取文本文件的键值对?

时间:2016-07-14 05:57:11

标签: python pandas dataframe key

我想解析一个包含以下数据的文本文件。

INPUT.TXT -

1=88|11=1438|15=KKK|45=7.7|45=00|21=66|86=a
4=13|4=1388|49=DDD|8=157.73|67=00|45=08|84=b|45=k
6=84|41=18|56=TTT|67=1.2|4=21|45=78|07=d

在此输入文本文件中,没有列被修复,可能是10或20或任何其他内容。我想用pandas解析这个文件。输出应包含:

output.txt的 -

index[0]
1     88
11    1438
15    kkk
45    7.7
45    00
21    66
86    a

index[1]
4     13
4     1388
49    DDD
8     157.73
67    00
45    08
84    b
45    k

关于如何获得此类结果的任何建议?

1 个答案:

答案 0 :(得分:3)

您可以先使用不在数据中的分隔符read_csv,例如;,然后使用split加倍stack

import pandas as pd
import numpy as np
import io

temp=u"""1=88|11=1438|15=KKK|45=7.7|45=00|21=66|86=a
4=13|4=1388|49=DDD|8=157.73|67=00|45=08|84=b|45=k
6=84|41=18|56=TTT|67=1.2|4=21|45=78|07=d
"""
#after testing replace io.StringIO(temp) to filename
df = pd.read_csv(io.StringIO(temp), sep=";", index_col=None, names=['text'])

print (df)
                                                text
0        1=88|11=1438|15=KKK|45=7.7|45=00|21=66|86=a
1  4=13|4=1388|49=DDD|8=157.73|67=00|45=08|84=b|45=k
2           6=84|41=18|56=TTT|67=1.2|4=21|45=78|07=d
s = df.text.str.split('|', expand=True).stack().str.split('=', expand=True)
print (s)
      0       1
0 0   1      88
  1  11    1438
  2  15     KKK
  3  45     7.7
  4  45      00
  5  21      66
  6  86       a
1 0   4      13
  1   4    1388
  2  49     DDD
  3   8  157.73
  4  67      00
  5  45      08
  6  84       b
  7  45       k
2 0   6      84
  1  41      18
  2  56     TTT
  3  67     1.2
  4   4      21
  5  45      78
  6  07       d
dfs = [g.set_index(0).rename_axis(None) for i, g in s.groupby(level=0)]
print (dfs[0])
       1
1     88
11  1438
15   KKK
45   7.7
45    00
21    66
86     a
for i, g in s.groupby(level=0):
    print (g.set_index(0).rename_axis(None))
       1
1     88
11  1438
15   KKK
45   7.7
45    00
21    66
86     a
         1
4       13
4     1388
49     DDD
8   157.73
67      00
45      08
84       b
45       k
      1
6    84
41   18
56  TTT
67  1.2
4    21
45   78
07    d

通过评论编辑:

如果需要写入文件s,请使用to_csv

s.to_csv('file.txt', header=None, index=None, sep='\t')  

EDIT1评论:

您可以将列名设置为空字符串,并按rename_axis删除索引名称(pandas 0.18.0中的新内容),但更常见的是将列名设置为某些文本(例如{{ 1}}):

s.columns = ['idx','a']