到目前为止我的代码:
from collections import OrderedDict as od
def load(fileIn, fileOut):
with open(fileIn+'.txt') as fin, open(fileOut+'.txt', 'w') as fout:
dict = od()
for line in fin:
row = line.split()
id = int(row[0])
frame = int(row[2])
rect = [int(row[3]),int(row[4]),int(row[5]),int(row[6])]
dict = {frame:[id, rect]}
fout.writelines(str(dict)+'\n')
从文本文件中读取,以特定方式对其进行排序并将其写入新文件。我需要添加另一个for
或两个循环,以便在编写它之前更好地对其进行排序,这就是我正在努力的地方。
以下是输入和输出示例,以使事情更清楚:
输入:
2 109 1 561 1 20 28 1
2 109 2 557 1 24 32 1
2 109 3 557 5 24 32 1
2 109 4 553 5 28 32 1
2 109 5 553 1 36 40 1
239 195 1 101 549 40 28 1
239 195 2 100 549 40 28 1
239 195 3 98 549 40 28 1
239 195 4 91 551 40 28 1
239 195 5 93 549 40 28 1
输出:
{1: [2, [561, 1, 20, 28]]}
{2: [2, [557, 1, 24, 32]]}
{3: [2, [557, 5, 24, 32]]}
{4: [2, [553, 5, 28, 32]]}
{5: [2, [553, 1, 36, 40]]}
{1: [239, [101, 549, 40, 28]]}
{2: [239, [100, 549, 40, 28]]}
{3: [239, [98, 549, 40, 28]]}
{4: [239, [91, 551, 40, 28]]}
{5: [239, [93, 549, 40, 28]]}
我试图将不同rect
s 的所有值归为一个键,这是frame
它们共享的共同点。因此,如果1每次都frame
在不同的文件中出现 100 次,我需要一个下的所有 s,其中将有 100 个不同的 s。id
rect
key
rect
所以一个例子是:
{1:[rect],[rect],[rect],[rect],[rect],[rect],[rect],[rect],[rect]}
{2:[rect],[rect],[rect],[rect],[rect],[rect],[rect],[rect],[rect]}
{3:[rect],[rect],[rect],[rect],[rect],[rect],[rect],[rect],[rect]}
然后我可以frame
1
在一个文件frame
1
中比较另一个文件。