Python：当第一列与列表中的字符串匹配时，从文本文件中拉线

Question

我有一个列表['dog', 'cat', 'snake', 'lizard'] 。 我想使用此列表从文本文件中提取行。 我的文本文件用制表符分隔，每行末尾用换行符分隔。 每行有4列，第一列是我列表中的名称之一。 前五行如下所示：

dog     data1     data2    data3
dog     data1     data2    data3
cat     data1     data2    data3
snake   data1     data2    data3
lizard  data1     data2    data3

许多行。

我想为列表中的每个项目创建一个文本文件。 在每个新文件中，我都希望原始文件中的每一行都与第一列匹配列表/新文件中的名称。 这是我编写的代码：

filename = "data.txt"
f = open(filename, 'r')

#my list is named Species
for names in Species:
    with open(str(names) + ".txt", 'w') as g:       
        for line in f:
            row = line.split()
            if names == row[0]:
                g.write(row)

我可以创建要写入的文本文件，但是没有写入任何文件。 我没有收到任何错误消息。 最后，我只希望为我有兴趣放入新文本文件中的每一行提取一些数据列。

Answer 1

尝试将列表直接写到文件中会导致错误（在Python中不合法）：

Python 2.7：

Python 2.7.10 (default, Sep 13 2015, 20:30:50) 
[GCC 5.2.1 20150911] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> with open("test", "w") as f:
...   f.write([1,2,3,4])
... 
Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
TypeError: expected a character buffer object
>>> 

未调用write操作，可能是因为没有与Species[0]相匹配的行。 当在Species[1]上再次调用顶层的for循环时， f已经在文件末尾，不再提供任何行。 seek在循环的开始文件的开头：

for name in Species:
    f.seek(0)
    with open(str(names) + ".txt", "w") as g:
        for line in f:
            if line.startswith(name):
                g.write(line)

另外（这是我要做的），您可以扫描一次f ，然后在处理时将每行分配给适当的动物：

records = {}
for line in f:
    animal = line.split()[0]
    if not records.get(animal):
        records[animal] = []
    records[animal].append(line)

for animal in records.keys():
    with open("{}.txt".format(animal), "w") as f:
        for line in records[animal]:
            f.write(line)

Answer 2

这是更新的代码！

Species = ['dog', 'cat', 'snake', 'lizard']
filename = "data.txt"
for names in Species:
    with open(str(names) + ".txt", 'w') as g:
        f = open(filename, 'r')
        for line in f:
            row = line.split()
            if names == row[0]:
                g.write(str(row))
        f.close()

1. 您需要在g.write（）中添加str（row），因为您无法将数组写入文本文件。
2. 重新打开“ data.txt”似乎可以解决文件未写入的问题（虽然不太确定，但是编辑：哦，这就是为什么:)）