Python:按预先设定的索引重新排列列表

时间:2017-12-17 00:57:05

标签: python python-3.x python-2.7 list python-2.x

所以我想用3行输入一个输入文件。 第一行是列表中的项目数 第二行,关于我如何重新排列列表的索引。 (更多信息见下文) 第三行,数字列表本身

因此,in.txt中的第二行是重新排列列表的索引。很难解释,但这是如何工作(见in.txt参考) 第一个数字保持不变,第二个数字变为第三个,第五个数字变为第二个,第四个变为第五个,等等。

这里有一些示例输入: in.txt

5
1 3 4 5 2
12 33 96 84 74

output.txt的

12
74
33
96
84

这是程序本身:

fin = open('in.txt','r') #file in
fout = open('output.txt','w') #file out

indata = fin.readlines()
cownum = int(indata[0])
print(cownum)
posmove = indata[1]
cid = indata[2]

poss=posmove.split()
ids=cid.split()

print(poss)
print("")

i=0
while i != int(cownum):
  print(poss[i])
  ids.insert((int(poss[i])-1), ids.pop(i))
  i=i+1
  print(i)
  print(ids)


newids = str(ids)
fout.write(newids)


fin.close()
fout.close()

仅为了一些背景信息,列表中的数字有点像产品ID&#39>

现在的问题是我重新安排了两次项目。

很抱歉,如果我有点不清楚(大笑和奇怪的变量名称)

3 个答案:

答案 0 :(得分:2)

将文件读入列表后,您可以使用以下代码对其进行排序:

>>> indexes = [1, 3, 4, 5, 2]
>>> values = [12, 33, 96, 84, 74]
>>> newlist = [0, 0, 0, 0, 0]
...
...
>>> for x, y in zip(indexes, values):
...     newlist[x-1] = y
... 
>>> for x in newlist:
...     print(x)
... 
12
74
33
96
84

答案 1 :(得分:2)

你可以试试这个:

file_data = [map(int, i.strip('\n').split()) for i in open('filename.txt')]
new_data = [i[-1] for i in sorted(zip(file_data[1], file_data[-1]), key=lambda x:x[0])]

输出:

[12, 74, 33, 96, 84]

答案 2 :(得分:1)

假设您已设法提取职位列表和产品ID列表(您仍需要将数据转换为整数):

positions = [1, 3, 4, 5, 2]
product_ids = [12, 33, 96, 84, 74]

您可以使用名为decorate-sort-undecorate的模式按给定位置对产品ID进行排序:

>>> [prod_id for (pos, prod_id) in sorted(zip(positions, product_ids))]
[12, 74, 33, 96, 84]

分步说明:

我们首先装饰产品ID列表及其对应的位置:

>>> zip(positions, product_ids)
[(1, 12), (3, 33), (4, 96), (5, 84), (2, 74)]

然后我们排序装饰列表 - 这将给我们一个正确的顺序,因为元组按项目顺序排序,并且由于第一个项目是一个位置,这给了我们想要的顺序。

>>> sorted(zip(positions, product_ids))
[(1, 12), (2, 74), (3, 33), (4, 96), (5, 84)]

最后,我们 undecorate (解包)现在排序的元组列表,以获得所需顺序的产品ID列表(同时丢弃其位置信息):

>>> [prod_id for (pos, prod_id) in sorted(zip(positions, product_ids))]
[12, 74, 33, 96, 84]

(使用Python 2进行演示)