将数据添加到列并附加它

Question

我在文件 input.txt 中有一个输入数据。我想在第 4 列的每一行添加 200，并希望以与原始格式相同的格式附加到它旁边。

输入.txt

>   >   >
0.000   5.064   -0.15835E-01 -7.755619582136471027e-03
0.700   9.064   -0.21715E-02 -1.302134466433525067e-02
0.800   5.064   0.18954E-01 -1.691809184964498253e-02
>   >   >
0.500   4.977   0.32899E-02 1.439701445411046415e-02
0.200   6.977   0.15972E-01 6.094966468334198084e-03
0.700   4.977   0.30631E-01 -5.640445453166963163e-03

out.txt

>   >   >
0.000   5.064   -0.15835E-01 -7.755619582136471027e-03 199.99224438041787
0.700   9.064   -0.21715E-02 -1.302134466433525067e-02 199.98697865533566
0.800   5.064   0.18954E-01 -1.691809184964498253e-02  199.98308190815035
>   >   >
0.500   4.977   0.32899E-02 1.439701445411046415e-02   200.01439701445412
0.200   6.977   0.15972E-01 6.094966468334198084e-03   200.00609496646834
0.700   4.977   0.30631E-01 -5.640445453166963163e-03  199.99435955454683

我试过了：

import numpy as np
data=np.loadtxt('input.txt')
data1=data(:,3)+200

我收到错误ValueError: could not convert string to float: '>' 我希望有专家能帮助我克服这个问题。

Answer 1

文件中有非数字符号。您可以将它们视为 cmets：

data = np.loadtxt('input.txt', comments=">")

但是您将无法将它们保存在输出文件中。恐怕您唯一的选择是在循环中逐行读取文件：

CHANGE = 200
with open('input.txt') as infile, open('out.txt', 'w') as outfile:
    for line in infile:
        if not line.startswith('>'): # Needs a modification
            parts = line.split()
            parts.append(str(float(parts[3]) + CHANGE) + '\n')
            line = ' '.join(parts)
        outfile.write(line)

Answer 2

我不太明白这个问题，但我想我知道问题出在哪里。我可以通过以下几种方式帮助您：

• 删除文件开头的>s。
• 跳过>s。像这样：

with open("input.txt", "r") as f:
    with open("temporaryfile.txt", "w") as temp:
        temp.write(f.read().replace(">   >   >",""))
    
    
data=np.loadtxt("temporaryfile.txt")
data1=data(:,3)+200

之后只需使用import os;os.remove("temporaryfile.txt") 删除temporaryfile.txt。

问题是它遇到了>s，这就是导致错误的原因。所以，如果你删除这些，那应该可以解决你的问题。

Answer 3

我们可以在不创建新文件的情况下实现这一点，并使用 fileinput 库就地执行操作。在打击代码中，它也保留了这个“>”字符。

import sys
import fileinput

file = fileinput.input('test.txt', inplace=1)

for i, line in enumerate(file):
    if line.startswith('>'):
        sys.stdout.write(line)
    else:
        string = line.split()
        string.append('200\n')
        line = ' '.join(string)
        sys.stdout.write(line)
file.close()

以上代码的输出：

>   >   >
0.700 4.977 0.30631E-01 -5.640445453166963163e-03 200
0.700 4.977 0.30631E-01 -5.640445453166963163e-03 200
0.700 4.977 0.30631E-01 -5.640445453166963163e-03 200
>   >   >
0.700 4.977 0.30631E-01 -5.640445453166963163e-03 200
0.700 4.977 0.30631E-01 -5.640445453166963163e-03 200
0.700 4.977 0.30631E-01 -5.640445453166963163e-03 200