从文件中读取并写入另一个python

时间:2018-05-20 13:24:44

标签: python file-handling

我的文件内容如下,

to-56  Olive  850.00  10 10
to-78  Sauce  950.00  25 20
to-65  Green  100.00   6 10

如果第4列数据小于或等于第5列,则应将数据写入第二个文件。
我尝试了以下代码,但只有'to-56 Olive'保存在第二个文件中。我无法弄清楚我在这里做错了什么。

file1=open("inventory.txt","r")
file2=open("purchasing.txt","w")
data=file1.readline()
for line in file1:

    items=data.strip()
    item=items.split()

    qty=int(item[3])
    reorder=int(item[4])

    if qty<=reorder:
        file2.write(item[0]+"\t"+item[1]+"\n")


file1.close()
file2.close()

2 个答案:

答案 0 :(得分:2)

我已经稍微更改了您的代码,您需要做的就是迭代文件中的行 - 就像这样:

file1=open("inventory.txt","r")
file2=open("purchasing.txt","w")

# Iterate over each line in the file
for line in file1.readlines():

    # Separate each item in the line
    items=line.split()

    # Retrieve important bits
    qty=int(items[3])
    reorder=int(items[4])

    # Write to the file if conditions are met
    if qty<=reorder:
        file2.write(items[0]+"\t"+items[1]+"\n")

# Release used resources
file1.close()
file2.close()

以下是 purchasing.txt 中的输出:

to-56   Olive
to-65   Green

答案 1 :(得分:1)

您只阅读一行输入。因此,您最多可以有一行输出。

我看到你的代码有点老了&#34;。这是一个更现代的&#34;和Pythonic版本。

# Modern way to open files. The closing in handled cleanly
with open('inventory.txt', mode='r') as in_file, \
     open('purchasing.txt', mode='w') as out_file:

    # A file is iterable
    # We can read each line with a simple for loop
    for line in in_file:

        # Tuple unpacking is more Pythonic and readable
        # than using indices
        ref, name, price, quantity, reorder = line.split()

        # Turn strings into integers
        quantity, reorder = int(quantity), int(reorder)

        if quantity <= reorder:
            # Use f-strings (Python 3) instead of concatenation
            out_file.write(f'{ref}\t{name}\n')