python - 使用 python 处理此文本文件的大多数 pythonic 方式

Question

我有一个包含测试数据的文本文件，如下所示:

hdr 1

hdr2

hdr3

data1

data2

data3

data1

data2

....

每行之间有一个空行。

我需要创建一个包含列表的列表

[[hdr1,hdr2,hdr3],[data1,data2,data3],[data1,data2,...]

什么是简洁的 pythonic 方式来做到这一点？

Answer 1

假设您的数据总是像您的示例中那样以 3 个为一组，您可能会使用 itertools，如果您将同一个迭代器传递 6 次，那么您最终应该得到 6 个字段，3其中是空行，其他是您感兴趣的 3 组:

import itertools

arr = []
with open('input.txt') as f:
    for field1,blank1,field2,blank2,field3,blank3 in itertools.izip_longest(*[f]*6):
        arr.append([field1,field2,field3])

(灵感来自this post)

编辑:您可能需要在字段上调用 ​​strip() 以便不包含任何额外的\n，因此 arr.append([field1 .strip(),field2.strip(),field3.strip()])