サブリストを含むリストが与えられた場合、指定された長さのサブリストを抽出したいと考えています。サブリストの長さが指定より短い場合は、すべて抽出します。明確にするために以下を参照してください
以下の例では、長さ = 2 のサブリストを抽出しています。長さがそれより大きい場合、サブリストの最初の 2 つの要素を抽出し、残りを無視します。入力
A = [['A',[1,2,3]],['D',[3,4]],['E',[6,7]],['F',[1]],['G',[7,6,5,4]]]
出力
B = [['A',[1,2]],['D',[3,4]],['E',[6,7]],['F',[1]],['G',[7,6]]]
私は現在次のようにやっています、それは動作しますが、簡単な方法があるかどうか疑問に思っています
B=[]
for el in A:
l = []
if len(el[1]) > 2:
l.append(el[0])
l.append(el[1][0:2])
B.append(l)
else:
l.append(el[0])
l.append(el[1][0:2])
B.append(l)
print B