python - 从 Python 中的文本文件中删除空白位置 (x,y)？

Question

我正在研究视频文件中的对象跟踪，我在视频的每一帧中有 5 到 6 个对象。由于代码中的一些错误，有些帧被跳过并在文本文件中显示为空白。如下图

5.0  2.3 5.4 5.2 5.5 nan nan 9.4 9.9 nan nan 
5.06 
5.13 2.8 5.6 6.2 6.6 9.45 8.7 6.2 7.8 9.9 5.7

正如您在这里看到的，第二帧 5.06 有一个空白行，其中我想要 nan。

我每个视频有 60K 帧，请你提供一些关于如何将空白区域更改为 nan 的建议。 60K帧手动改真的很繁琐。提前致谢 。预期的输出是

5.0  2.3 5.4 5.2 5.5 nan nan 9.4 9.9 nan nan
5.06 nan nan nan nan nan nan nan nan nan nan
5.13 2.8 5.6 6.2 6.6 9.45 8.7 6.2 7.8 9.9 5.7

Answer 1

如果这是一个简单的文本文件，并且每个 frame 都在一个新行中。您可以简单地创建一个包含行中只有 1 个元素的框架的新文件，并向这些行添加 12 nan。示例代码 -

with open('input.txt','r') as f , open('newoutput.txt','w') as oF:
    for line in f:
        eles = line.split() # this splits the line by whitespaces
        if len(eles) == 1:
            line = eles[0] + " nan" * 12 + "\n"
        oF.write(line)

我给上面的程序输入了以下内容-

5.0  2.3 5.4 5.2 5.5 nan nan 9.4 9.9 nan nan 
5.06 
5.13 2.8 5.6 6.2 6.6 9.45 8.7 6.2 7.8 9.9 5.7

它给了我输出-

5.0  2.3 5.4 5.2 5.5 nan nan 9.4 9.9 nan nan 
5.06 nan nan nan nan nan nan nan nan nan nan nan nan
5.13 2.8 5.6 6.2 6.6 9.45 8.7 6.2 7.8 9.9 5.7