Question

我正在尝试学习Python，但我被困在这里，任何帮助都会受到赞赏。

我有2个文件。

1是一个.dat文件，没有列标题是固定宽度，包含多行数据 1是包含列标题，列长度和数据类型

的.fmt文件

.dat示例：

10IFKDHGHS34
12IFKDHGHH35
53IFHDHGDF33

.fmt示例：

ID,2,n
NAME,8,c
CODE,2,n

所需输出为.csv：

ID,NAME,CODE
10,IFKDHGHS,34
12,IFKDHGHH,35
53,IFHDHGDF,33

Answer 1

首先，我要解析格式文件。

with open("format_file.fmt") as f:
    # csv.reader parses away the commas for us 
    # and we get rows as nice lists
    reader = csv.reader(f) 
    # this will give us a list of lists that looks like
    # [["ID", "2", "n"], ...]
    row_definitions = list(reader)

# iterate and just unpack the headers
# this gives us ["ID", "NAME", "CODE"]
header = [name for name, length, dtype in row_definitions]
# [2, 8, 2]
lengths = [int(length) for name, length, dtype in row_definitions]

# define a generator (possibly as a closure) that simply slices 
# into the string bit by bit -- this yields, for example, first 
# characters 0 - 2, then characters 2 - 10, then 8 - 12
def parse_line(line):
    start = 0
    for length in lengths:
        yield line[start: start+length]
        start = start + length

with open(data_file) as f:
     # iterating over a file pointer gives us each line separately
     # we call list on each use of parse_line to force the generator to a list
     parsed_lines = [list(parse_line(line)) for line in data_file]

# prepend the header
table = [header] + parsed_lines

# use the csv module again to easily output to csv
with open("my_output_file.csv", 'w') as f:
    writer = csv.writer(f)
    writer.writerows(table)

使用Python或Python / Pandas以编程方式将带有.fmt文件的固定宽度.data文件转换为.csv文件

1 个答案: