2

次の単純化されたケースを考えてみましょう。

lol = [['John','Polak',5,3,7,9],
       ['John','Polak',7,9,2,3],
       ['Mark','Eden' ,0,3,3,1],
       ['Mark','Eden' ,5,1,2,9]]

最初の2つのパラメーターに基づいて、このリストのリストをリストのリストのリストに変換するためのpythonicでメモリ+速度の効率的な方法は次のとおりです。

lolol = [[['John','Polak',5,3,7,9],
          ['John','Polak',7,9,2,3]],
         [['Mark','Eden' ,0,3,3,1],
          ['Mark','Eden' ,5,1,2,9]]]

実際には、正しい階層がある限り、他のデータ構造でも問題ありません。たとえば、次の辞書構造が思い浮かびますが、それを作成しても速度効率が十分ではないようで、メモリはおそらく lolol ソリューションよりも高くなります。

dolol = {('John','Polak'):[[5,3,7,9],[7,9,2,3]],
         ('Mark','Eden') :[[0,3,3,1],[5,1,2,9]]}
4

3 に答える 3

6

リスト：

from itertools import groupby
lolol = [list(grp) for (match, grp) in groupby(lol, lambda lst: lst[:2])]
# [[['John', 'Polak', 5, 3, 7, 9], ['John', 'Polak', 7, 9, 2, 3]],
#  [['Mark', 'Eden', 0, 3, 3, 1], ['Mark', 'Eden', 5, 1, 2, 9]]]

辞書：

dolol = dict((tuple(match), [x[2:] for x in grp]) for (match, grp) in 
             groupby(lol, lambda lst: lst[:2]))
# {('John', 'Polak'): [[5, 3, 7, 9], [7, 9, 2, 3]],
#  ('Mark', 'Eden'): [[0, 3, 3, 1], [5, 1, 2, 9]]}

itertools.groupby連続した一致で機能するため、ソートされた入力 ( ) を想定していますlol。

于 2011-02-12T21:59:39.227 に答える
5

辞書が受け入れられる場合、次のコードは辞書を作成します。

import collections
d = collections.defaultdict(list)
for name, surname, *stuff in lol:
    d[name, surname].append(nums)

これには Python 3 (拡張された iterable unpacking ) が必要であることに注意してください。Python 2 の場合は、

for x in lol:
    name = x[0]
    surname = x[1]
    stuff = x[2:]

行を節約するために変数を折りたたむことができます。

于 2011-02-12T20:30:31.823 に答える
0

デルナンの答えをPython2に相当するもので補完するには：

from collections import defaultdict

dolol=defaultdict(list)
for data in lol:
    dolol[data[0],data[1]].append(data[2:])
于 2011-02-13T07:29:18.197 に答える