1

データベース行の値を表す同じ長さの n 個のリストがあります。データはかなり複雑なので、例では単純化した値を示します。

基本的に、これらのリスト (a、b、c) の値を、キーがリスト (id) のセットである辞書にマップしたいと考えています。

リストの例:

id = [1,1,1,2,2,2,3,3,3]
a = [1,2,3,4,5,6,7,8,9]
b = [10,11,12,13,14,15,16,17,18]
c = [20,21,22,23,24,25,26,27,28]

必要な辞書出力:

{id:[[a],[b],[c]],...}

{'1':[[1,2,3],[10,11,12],[20,21,22]],'2':[[4,5,6],[13,14,15],[23,24,25]],'3':[[7,8,9],[16,17,18],[26,27,28]]}

ディクショナリには、現在ディクショナリ キーである id リストの一意の値によってサブセット化された、元の a、b、c の値のリストのリストがあります。

これが十分に明確であることを願っています。

4

2 に答える 2

1

これを試して:

id = ['1','1','1','2','2','2','3','3','3']
a  = [1,2,3,4,5,6,7,8,9]
b  = [10,11,12,13,14,15,16,17,18]
c  = [20,21,22,23,24,25,26,27,28]

from collections import defaultdict
d = defaultdict(list)

# add as many lists as needed, here n == 3
lsts = [a, b, c]

for ki, kl in zip(id, zip(*lsts)):
    d[ki] += [kl]

for k, v in d.items():
    # if you don't mind using tuples, simply do this: d[k] = zip(*v)
    d[k] = map(list, zip(*v))

質問によると、結果はまさに期待どおりです。

d == {'1':[[1,2,3],[10,11,12],[20,21,22]],
      '2':[[4,5,6],[13,14,15],[23,24,25]],
      '3':[[7,8,9],[16,17,18],[26,27,28]]}
=> True
于 2013-08-21T22:54:09.280 に答える
0
IDs = [1,1,1,2,2,2,3,3,3]
a = [1,2,3,4,5,6,7,8,9]
b = [10,11,12,13,14,15,16,17,18]
c = [20,21,22,23,24,25,26,27,28]

import itertools
d = {}
for key, group in itertools.groupby(sorted(zip(IDs, a, b, c)), key=lambda x:x[0]):
    d[key] = map(list, zip(*group)[1:])  # [1:] to get rid of the ID

print d

出力:

{1: [[1, 2, 3], [10, 11, 12], [20, 21, 22]], 
 2: [[4, 5, 6], [13, 14, 15], [23, 24, 25]], 
 3: [[7, 8, 9], [16, 17, 18], [26, 27, 28]]}
于 2013-08-21T22:55:03.830 に答える