按不同值分组

Question

我有一个值列表，比如元组，比如

    tmis = [
           ( 'script1', 'function1', 'lock1' ),
           ( 'script1', 'function2', 'lock1' ),
           ( 'script1', 'function3', 'lock1' ),
           ( 'script2', 'function4', 'lock1' ),
           ( 'script3', 'function5', 'lock2' ),
           ( 'script4', 'function6', 'lock3' ),
           ( 'script5', 'function7', 'lock3' ),
           ( 'script8', 'function10', 'lock1, lock2' )
        ]

我想用相同的“脚本”但不同的“锁”对数据进行分组。你知道吗

期望输出：

[ 
  [('script1', 'function1', 'lock1'), 
   ('script1', 'function2', 'lock1'), 
   ('script1', 'function3', 'lock1'), 
   ('script3', 'function5', 'lock2'), 
   ('script4', 'function6', 'lock3')],
  [('script2', 'function4', 'lock1'), 
   ('script5', 'function7', 'lock3')],
  [('script8', 'function10', 'lock1, lock2')]
]

因此，我对这些数据的输出是3个组，其中每个组只有一个特定的“锁”，但如果它们有相同的“脚本”值，那么它们可以放在同一个组中，而不管同一个“锁”的多次出现。另外，最后一个数据项“script8”有2个“locks”，因此应该位于不具有现有“lock1”或“lock2”的组中。你知道吗

它不必是这种格式，我可以使用dictionary/set/tuples/list/任何其他格式。我找到了一个方法，但它并不漂亮，所以我想知道是否有一个简单的方法来做这个分组。你知道吗

我试过（从另一个解决方案中发现）

groups = []
uniquekeys = []
data = sorted(tmis, key=lambda x: x[0] )
for k, g in itertools.groupby(data, lambda x: x[0]):
    groups.append( list(g) )
    uniquekeys.append(k)

但这个组只按“script”值进行分组。有什么想法吗？你知道吗

编辑：我想我的解释不是很清楚。我要做的是对元组进行分组，使具有相同“锁”的两个“脚本”不在同一个组中（除非“脚本”相同）

我可以张贴我目前（丑陋）的解决方案，如果它有帮助。你知道吗

Answer 1

以下是我的解决方案：

def getscript(group, lock):
    """
    returns the script used by the given lock in the given group
    or None if no script uses the given lock
    """
    for script, function, locks in group:
        if lock in locks.split(', '):
            return script
    return None

def compatible(group, task):
    """
    tells if a task ((script, function, locks) tuple) can be added to a group
    """
    for lock in task[2].split(', '):
        script = getscript(group, lock)
        if script and script != task[0]:
            return False
    return True

def makegroups(tasks):
    groups = []
    for task in tasks:
        try:
            group = next(grp for grp in groups if compatible(grp, task))
        except StopIteration:
            groups.append([task])
        else:
            group.append(task)
    return groups

Answer 2

你的例子与你所描述的不符。例如，s3、l2和s4、l3与s1、l1在同一组中

假设你只是想做你在课文中说的，而不是你想要的输出。这是密码-

tmis = [
           ( 's1', 'm1', 'l1' ),
           ( 's1', 'm2', 'l1' ),
           ( 's1', 'm3', 'l1' ),
           ( 's2', 'm4', 'l1' ),
           ( 's1', 'm5', 'l2' ),
           ( 's4', 'm6', 'l3' ),
           ( 's5', 'm7', 'l3' ),
           ( 's8', 'm10', 'l1, l2' )
        ]

d={}

for i in tmis:
        flag=0
        if d.has_key(i[2]):
            d[i[2]].append(i)
            flag=1
        else:
            for key in d.keys():
                if d[key][0][0] == i[0]:
                    d[key].append(i)
                    flag=1
                    break

        if flag==0:
            d[i[2]] = []
            d[i[2]].append(i)

print d