这里的问题是你正在使用readlines()
两次,意思是第一次读取数据,第二次就什么都没有了。
您可以直接迭代文件而不使用readlines()
- 事实上,这是更好的方法,因为它不会立即读取整个文件。
虽然您可以通过使用来做您想做的事情str.split()
正如你所拥有的,更好的选择是使用the csv module,这是专为该任务而设计的。
import csv
with open("addressbook1.txt") as input, open("college_age.txt", "w") as college, open("adult_age.txt", "w") as adult:
reader = csv.DictReader(input, dialect="excel-tab")
fieldnames = reader.fieldnames
writer_college = csv.DictWriter(college, fieldnames, dialect="excel-tab")
writer_adult = csv.DictWriter(adult, fieldnames, dialect="excel-tab")
writer_college.writeheader()
writer_adult.writeheader()
for row in reader:
if int(row["Age"]) < 23:
writer_college.writerow(row)
else:
writer_adult.writerow(row)
那么我们在这里做什么?首先我们使用the with陈述 for 打开文件。它不仅更具Python风格和可读性,而且可以为您处理关闭,即使发生异常也是如此。
接下来我们创建一个DictReader
从文件中读取行作为字典,自动使用第一行作为字段名称。然后,我们让编写器写回我们的拆分文件,并写入标头。使用DictReader
这是一个偏好问题。它通常在您经常访问数据的情况下使用更多(并且当您不知道列的顺序时),但它使代码在这里易于阅读。但是,您可以只使用标准csv.reader()
.
接下来,我们循环遍历文件中的行,检查年龄(我们将其转换为 int,以便我们可以进行数值比较)以了解要写入哪个文件。这with
声明为我们关闭了文件。
对于多个输入文件:
import csv
fieldnames = ["Name", "Date", "Age", "Sex", "Color"]
filenames = ["addressbook1.txt", "addressbook2.txt", ...]
with open("college_age.txt", "w") as college, open("adult_age.txt", "w") as adult:
writer_college = csv.DictWriter(college, fieldnames, dialect="excel-tab")
writer_adult = csv.DictWriter(adult, fieldnames, dialect="excel-tab")
writer_college.writeheader()
writer_adult.writeheader()
for filename in filenames:
with open(filename, "r") as input:
reader = csv.DictReader(input, dialect="excel-tab")
for row in reader:
if int(row["Age"]) < 23:
writer_college.writerow(row)
else:
writer_adult.writerow(row)
我们只需添加一个循环来处理多个文件。请注意,我还添加了字段名称列表。在我只使用文件中的字段和顺序之前,但由于我们有多个文件,我认为在这里这样做会更明智。另一种方法是使用第一个文件来获取字段名称。