我有一个递归函数,它从一个文件中读取一个scout记录列表,然后按ID的顺序添加到一个列表框中使用addScouts(1)
调用函数,函数如下:
def addScouts(self,I):
i = I
with open(fileName,"r") as f:
lines = f.readlines()
for line in lines:
if str(line.split(",")[3])[:-1] == str(i):
self.scoutList.insert(END,line[:-1])
i += 1
return self.addScouts(i)
return
我的问题是我的文件ID被排序为
1,2,4,5
,因为在某个时候我删除了ID为3的scout但是,当我在列表框(上面的函数)中运行重新排序童子军的函数时,它只列出ID 3及以下的童子军。这是因为当i = 3
时,文件中的任何项都不等于3,因此函数到达末尾并在有机会检查剩余记录之前返回。文件内容:
Kris,Rice,17,1
Olly,Fallows,17,2
Olivia,Bird,17,4
Louis,Martin,18,5
你知道怎么解决这个问题吗?
最佳答案
只需在最后一列排序:
sorted(f,key=lambda x: int(x.split(",")[-1]))
您可以使用“等分”查找新数据的放置位置,以便在对其排序一次后保持数据的顺序:
from bisect import bisect
import csv
with open("foo.txt") as f:
r = list(csv.reader(f))
keys = [int(row[-1]) for row in r]
new = ["foo","bar","12","3"]
ind = bisect(keys, int(new[-1]))
r.insert(ind,new)
print(r)
输出:
[['Kris', 'Rice', '17', '1'], ['Olly', 'Fallows', '17', '2'], ['foo', 'bar', '12', '3'], ['Olivia', 'Bird', '17', '4'], ['Louis', 'Martin', '18', '5']]
一种更简单的方法是检查第一行是否具有更高的id,如果没有更高的id,则将其追加到末尾:
import csv
with open("foo.txt") as f:
r = list(csv.reader(f))
new = ["foo","bar","12","3"]
key = int(new[-1])
ind = None
for i, row in enumerate(r):
if int(row[-1]) >= key:
ind = i
break
r.insert(ind, new) if ind is not None else r.append(new)
print(r)
输出:
[['Kris', 'Rice', '17', '1'], ['Olly', 'Fallows', '17', '2'], ['foo', 'bar', '12', '3'], ['Olivia', 'Bird', '17', '4'], ['Louis', 'Martin', '18', '5']
为了在添加新值时始终保持该文件的顺序,我们只需写入临时文件,将行写入正确的位置,然后用更新的文件替换原始文件:
import csv
from tempfile import NamedTemporaryFile
from shutil import move
with open("foo.csv") as f, NamedTemporaryFile(dir=".", delete=False) as temp:
r = csv.reader(f)
wr = csv.writer(temp)
new = ["foo", "bar", "12", "3"]
key, ind = int(new[-1]), None
for i, row in enumerate(r):
if int(row[-1]) >= key:
wr.writerow(new)
wr.writerow(row)
wr.writerows(r)
break
wr.writerow(row)
else:
wr.writerow(new)
move(temp.name, "foo.csv")
foo.csv之后将按顺序显示数据:
Kris,Rice,17,1
Olly,Fallows,17,2
foo,bar,12,3
Olivia,Bird,17,4
Louis,Martin,18,5
关于python - 当调用递归函数对值进行排序时,它会遗漏一个。我该如何解决?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/30968243/