文字列のリストがあり、リスト内の単語の文字長に基づいて、そのリストを異なる「サブリスト」に分割したいと考えています。
List = [a, bb, aa, ccc, dddd]
Sublist1 = [a]
Sublist2= [bb, aa]
Sublist3= [ccc]
Sublist2= [dddd]
どうすればPythonでこれを達成できますか?
ありがとうございました
文字列のリストがあり、リスト内の単語の文字長に基づいて、そのリストを異なる「サブリスト」に分割したいと考えています。
List = [a, bb, aa, ccc, dddd]
Sublist1 = [a]
Sublist2= [bb, aa]
Sublist3= [ccc]
Sublist2= [dddd]
どうすればPythonでこれを達成できますか?
ありがとうございました
を使用してitertools.groupby
:
values = ['a', 'bb', 'aa', 'ccc', 'dddd', 'eee']
from itertools import groupby
output = [list(group) for key,group in groupby(sorted(values, key=len), key=len)]
結果は次のとおりです。
[['a'], ['bb', 'aa'], ['ccc', 'eee'], ['dddd']]
リストがすでに文字列の長さでソートされていて、グループ化するだけでよい場合は、コードを次のように単純化できます。
output = [list(group) for key,group in groupby(values, key=len)]
辞書を使うべきだと思います
>>> dict_sublist = {}
>>> for el in List:
... dict_sublist.setdefault(len(el), []).append(el)
...
>>> dict_sublist
{1: ['a'], 2: ['bb', 'aa'], 3: ['ccc'], 4: ['dddd']}
>>> from collections import defaultdict
>>> l = ["a", "bb", "aa", "ccc", "dddd"]
>>> d = defaultdict(list)
>>> for elem in l:
... d[len(elem)].append(elem)
...
>>> sublists = list(d.values())
>>> print(sublists)
[['a'], ['bb', 'aa'], ['ccc'], ['dddd']]
長さでインデックス付けされたリストのリストに満足していると仮定すると、次のようなものはどうですか
by_length = []
for word in List:
wl = len(word)
while len(by_length) < wl:
by_length.append([])
by_length[wl].append(word)
print "The words of length 3 are %s" % by_length[3]