替换python文件中每行的第一个字符

Question

我试图替换多个txt文件行中的前几个字符，并使用此代码，但输出错误：

for filename in glob.glob('./labels/**/*.txt', recursive=True):
with open(filename, 'r') as f:
    original_lines = f.readlines()
with open(filename, 'w') as out:
    for line in original_lines:
        if line.startswith('0'):
            line0 = line
            out.write(line0)
        if line.startswith('1'):
            line1 = line
            out.write(line1)
        if line.startswith('2'):
            line2 = line
            out.write(line2)
        if line.startswith('3'):
            line3 = line
            out.write(line3)
        if line.startswith('5'):
            line4 = '4' + line[1:]
            out.write(line4)
        if line.startswith('7'):
            line5 = '5' + line[1:]
            out.write(line5)
        if line.startswith('9'):
            line6 = '6' + line[1:]
            out.write(line6)
        if line.startswith('10'):
            line7 = '7' + line[1:]
            out.write(line7)
        if line.startswith('11'):
            line8 = '8' + line[1:]
            out.write(line8)
        if line.startswith('12'):
            line9 = '9' + line[1:]
            out.write(line9)

如果我有这样一个文件：

0.2 0.4 0.8

120.10.1 0.25

7 0.66 0.80 0.91

我希望输出为：

0.2 0.4 0.8

9 0.1 0.1 0.25

50.66 0.80 0.91

Answer 1

我认为您有一些切片问题，例如，假设您的输入是12 0.1 0.1 0.25line[1:]将是2 0.1 0.1 0.25，因为您的输入是字符串。您可以使用以下内容：

line = '12 0.1 0.1 0.25'.split(' ') #convert your string to a list

temp = int(line[0]) #get first element and convert to integer to make comparisons easier

if temp < 3:
    print(' '.join(line))
elif temp == 5: 
    line[0] = '4'
    print(' '.join(line))
elif temp == 7:
    line[0] = '5'
    print(' '.join(line))
elif temp > 8:
    line[0] =str(temp - 3)
    print(' '.join(line))

#Output: 

9 0.1 0.1 0.25

注意：在您的案例中，最好使用elif而不是if，因为如果您的一个条件为true，它将不会检查其余的条件。更多信息here