python读取table文件

2020-01-07 16:10:24 浏览数 (1)

有个table文件, 有时候需要处理header , 可以用linecache 模块

代码语言:javascript复制
#!/usr/bin/env python
# -*- coding: ascii -*-
import linecache
import fileinput
import sys
from collections import defaultdict
inputFile = sys.argv[1]
headerLine = linecache.getline(inputFile, 1).strip()
#print(headerLine)
Probenames = headerLine.split("t")[1:]
inputH = open(inputFile, "r")

d = defaultdict(list)
for line in inputH:
    if "Sample" not in line:
        z = line.rstrip().split("t")[1:]
        for num, p_data in enumerate(z):
            if p_data != "":
                d[Probenames[num]].append(p_data)
inputH.close()
print("NameProbetdata")
for p in d:
    for x in d[p]:
        #print(x, d[p])
        print("{0}t{1}".format(p, x))

当然也可以用 fileinput 模块

参考:  https://docs.python.org/3/library/fileinput.html#fileinput.isfirstline

0 人点赞