说我有分数:
points = [(1., 1.), (3., 0.), (-1., -1.), (9., 2.), (-4., 2.) ]
如果我按y轴排序:
points = sorted(points , key=lambda k: [k[1], k[0]])
我得到了
points = [(-1., -1.), (3., 0.), (1.,1.) , (-4.,2.), (9., 2.)]
但是我想将它完全独立于x轴排序。 此外,我希望输出为2个列表,显示两种可能的排序(即y值相等的x值的所有排列):
[(-1., -1.), (3., 0.), (1.,1.) , (-4.,2.),(9., 2.)]
[(-1., -1.), (3., 0.), (1.,1.) , (9.,2.), (-4.,2.)]
我有办法做到这一点吗?
答案 0 :(得分:2)
在给定等价关系的情况下(例如比较y坐标和忽略x坐标)创建所有可能的排序排列的多个列表:
以下是一些解决问题的工作代码:
from operator import itemgetter
from itertools import groupby, product, permutations, chain
points = [(1., 1.), (3., 0.),(-1., -1.) , (9., 2.), (-4., 2.) ]
points.sort(key=itemgetter(1))
groups = [list(permutations(g)) for k, g in groupby(points, itemgetter(1))]
for t in product(*groups):
print(list(chain.from_iterable(t)))
[(-1.0, -1.0), (3.0, 0.0), (1.0, 1.0), (9.0, 2.0), (-4.0, 2.0)]
[(-1.0, -1.0), (3.0, 0.0), (1.0, 1.0), (-4.0, 2.0), (9.0, 2.0)]
初始排序仅按y轴排序点。这使用itemgetter()来提取字段1.
groupby()步骤会生成具有相同y坐标的点组。
permutations()步骤会生成每个组的所有可能排序。
product()步骤生成每个排列组的笛卡尔积(这样每个输出都有一个来自每个排列组的元素)。
chain.from_iterable()步骤将产品中的连续元组链接到一个迭代中,可以将其输入list()以获得所需的结果。
1)按y坐标对点进行排序,忽略x坐标:
>>> points = [(1., 1.), (3., 0.),(-1., -1.) , (9., 2.), (-4., 2.)]
>>> points.sort(key=itemgetter(1))
>>> points
[(-1.0, -1.0), (3.0, 0.0), (1.0, 1.0), (9.0, 2.0), (-4.0, 2.0)]
>>> ^-----------^-----------^-----------^-------------^ ascending y-values
2)创建具有相同y坐标的点组:
>>> pprint([list(g) for k, g in groupby(points, itemgetter(1))], width=40)
[[(-1.0, -1.0)], # y = -1.0
[(3.0, 0.0)], # y = 0.0
[(1.0, 1.0)], # y = 1.0
[(9.0, 2.0), (-4.0, 2.0)]] # y = 2.0
3)生成具有相同y坐标的点的所有排列:
>>> groups = [list(permutations(g)) for k, g in groupby(points, itemgetter(1))]
>>> pprint(groups)
[[((-1.0, -1.0),)], # y = -1.0
[((3.0, 0.0),)], # y = 0.0
[((1.0, 1.0),)], # y = 1.0
[((9.0, 2.0), (-4.0, 2.0)), ((-4.0, 2.0), (9.0, 2.0))]] # y = 2.0
4)使用每个排列组中的一个元素创建所有可能的序列:
>>> for t in product(*groups):
print(t)
(((-1.0, -1.0),), ((3.0, 0.0),), ((1.0, 1.0),), ((9.0, 2.0), (-4.0, 2.0)))
(((-1.0, -1.0),), ((3.0, 0.0),), ((1.0, 1.0),), ((-4.0, 2.0), (9.0, 2.0)))
5)将每个子序列合并为一个列表:
>>> for t in product(*groups):
list(chain.from_iterable(t))
[(-1.0, -1.0), (3.0, 0.0), (1.0, 1.0), (9.0, 2.0), (-4.0, 2.0)]
[(-1.0, -1.0), (3.0, 0.0), (1.0, 1.0), (-4.0, 2.0), (9.0, 2.0)]
答案 1 :(得分:0)
仅对x值进行排序:
points = sorted(points , key=lambda k: k[1])
points
[(-1.0, -1.0), (3.0, 0.0), (1.0, 1.0), (9.0, 2.0), (-4.0, 2.0)]