繁体   English   中英

如何在过滤出某些值的同时对列表列表进行排序?

[英]How do I go about simultaneously sorting a list of lists while filtering out certain values?

我有一个三列的ascii文件,我想根据第三列对文件进行排序和过滤。 也就是说,我要从最小到最大排列第三列,并取出上面的所有值,例如8.0。 如果直接针对一个列表,我知道该怎么做。 但是,我不知道如何将对第三列所做的更改应用于所有列,因为每一行都相互链接。

小问题:我已经这样解压缩了ascii文件:

f=open('textfile.dat',"r")
lines=f.readlines()
result=[]
for x in lines:
    result.append(x.split('\n')[0])

a = []
b = []
c = []    
for w in result:
    ra.append(w.split()[0])
for x in result:
    dec.append(x.split()[1])
for y in result:
    mag.append(y.split()[2])

p=0
for i in a:
    a[p] = float(i)
    p= p+1    
q=0
for j in b:
    b[q] = float(j)
    q= q+1    
r=0
for k in c:
    c[r] = float(k)
    r= r+1

test=[a,b,c]

至少可以说这似乎不必要地乏味。 有什么办法可以更紧凑地做到这一点? 它只是一个具有四列的ascii文件。 尽管它们是数字,但它们被读为'str',因此我必须将其改回浮点数。

排序似乎只是使用转座子,然后根据列进行排序,这解决了我的第一个问题。

tset = zip(*test)
tset.sort(key = lambda x: x[2])
sorttest = zip(*tset)

但是,如果我想摆脱上面第三列中的所有数字,例如8.0,我会做类似的事情:

testrange[2] = [i for i in sorttest[2] if i <= 8.0]

但是,这仅适用于一行。 例如,我做不到

testrange[0] = [i*1.0 for i in sorttest[2] if i <= 8.0]
testrange[1] = [i*1.0 for i in sorttest[2] if i <= 8.0]

因此,如何在过滤出某些值的同时对列表列表进行排序?

编辑:

Sample input:
 [5,12,3,7,90]
 [9,1,63,23,8]
 [73,11,8,2,5]

Sample output:
 [7,90,3,12]
 [23,8,63,1]
 [2,5,8,11]

您可以使用numpy:

import numpy as np

x = [[5,12,3,7,90],
     [9,1,63,23,8],
     [73,11,8,2,5]]
x = np.array(x)
x = x[:, np.where(x[2] < 70)[0]]
x = x[:, np.argsort(x[2])].tolist()

那会给你:

[[7、90、3、12],[23、8、63、1],[2、5、8、11]

因此,我可能没有满足您的要求,但是对于每一行,您都可以列出该行拆分的项目列表。 您将得到列表的清单(行)(清单的一行中的单独项目)。 然后,您可以按每行的第3列对行进行排序。 然后按3s列值过滤该行列表。 我已经将这些作为单独的步骤完成,但是您可以将它们合并。

my_data = """1.4 2.5 5.6
2.4 7.5 9.8
4.8 9.7 2.5
4.5 6.5 7.9
1.3 3.4 12.6"""

list_of_data = [line.split() for line in my_data.split("\n")]
sorted_list = sorted(list_of_data, key=lambda line: float(line[2]))
filtered_list = [line for line in sorted_list if float(line[2]) < 8.0]
for line in filtered_list:
    print(" ".join(line))

输出值

4.8 9.7 2.5
1.4 2.5 5.6
4.5 6.5 7.9

因此,输出已删除了第三列等于或大于8.0的那些行。 并且同一行中的所有相关数字也仍然匹配,并且以第3列为基础

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM