python - 替换文件内的行中的字符

Question

我在文本文件中有这些不同的行和值

sample1:1
sample2:1
sample3:0
sample4:15
sample5:500

我希望有时更新“：”之后的数字我知道我可以用“：”拆分名称并获得一个包含 2 个值的列表。

f = open("test.txt","r")
lines = f.readlines()
lineSplit = lines[0].split(":",1)
lineSplit[1] #this is the value I want to change

我不太确定如何使用写入函数更新 lineSplit[1] 值

Answer 1

也许是这样的（假设之前:的每个第一个元素确实是一个键）：

from collections import OrderedDict

with open('fin') as fin:
    samples = OrderedDict(line.split(':', 1) for line in fin)

samples['sample3'] = 'something else'

with open('output') as fout:
    lines = (':'.join(el) + '\n' for el in samples.iteritems())
    fout.writelines(lines)

Answer 2

另一种选择是使用csv模块（:在您的情况下是列分隔符）。

假设有一个test.txt文件内容如下：

sample1:1
sample2:1
sample3:0
sample4:15
sample5:500

你需要增加每个值。以下是您的操作方法：

import csv


# read the file
with open('test.txt', 'r') as f:
    reader = csv.reader(f, delimiter=":")
    lines = [line for line in reader]


# write the file
with open('test.txt', 'w') as f:
    writer = csv.writer(f, delimiter=":")
    for line in lines:
        # edit the data here

        # e.g. increment each value
        line[1] = int(line[1]) + 1
    writer.writerows(lines)

现在的内容test.txt是：

sample1:2
sample2:2
sample3:1
sample4:16
sample5:501

但是，无论如何，fileinput在您的情况下使用听起来更合乎逻辑（编辑相同的文件）。

希望有帮助。

Answer 3

字符串是不可变的，这意味着您不能通过索引在其中分配新值。但是您可以将整个文件拆分为行列表，并完全更改单个行（字符串）。这就是你在 lineSplit[1] = A_NEW_INTEGER 中所做的

with open(filename, 'r') as f:
    lines = f.read().splitlines()
for i, line in enumerate(lines):
    if condition:
        lineSplit = line.split(':')
        lineSplit[1] = new_integer
        lines[i] = ':'.join(lineSplit)
with open(filename, 'w') as f:
    f.write('\n'.join(lines)

Answer 4

fileinput如果您尝试修改同一个文件，则可以使用该模块：

>>> strs = "sample4:15"

利用序列解包的优势将拆分后的结果存储在变量中。

>>> sample, value = strs.split(':')
>>> sample
'sample4'
>>> value
'15'

代码：

import fileinput
for line in fileinput.input(filename, inplace = True):
    sample, value = line.split(':')
    value = int(value)     #convert value to int for calculation purpose
    if some_condition: 
           # do some calculations on sample and value
           # modify sample, value if required 

    #now the write the data(either modified or still the old one) to back to file
    print "{}:{}".format(sample, value)