Python-1Dリストを可変長の2Dリストに変換します

Pythonのリストは通常​​、要素が次々にリストされる1Dリストです。しかし、2Dリストでは、外側のリスト内にネストされたリストがあります。この記事では、特定の1Dリストから2Dリストを作成する方法を説明します。また、2Dリスト内の要素数の値をプログラムに提供します。

追加とインデックスの使用

このアプローチでは、2Dリストの各要素をループするforループを作成し、それを作成する新しいリストのインデックスとして使用します。インデックス値をゼロから開始し、2Dリストから受け取る要素に追加することでインデックス値を増やし続けます。

例

# Given list
listA = ['Sun','Mon','Tue','Wed','Thu','Fri','Sat']

# Length of 2D lists needed
len_2d = [ 2, 4]

#Declare empty new list
res = []
def convert(listA, len_2d):
   idx = 0
   for var_len in len_2d:
      res.append(listA[idx: idx + var_len])
      idx += var_len
convert(listA, len_2d)
print("The new 2D lis is: \n",res)

上記のコードを実行すると、次の結果が得られます-

出力

The new 2D lis is:
[[1, 2], [3, 4, 5, 6]]

isliceの使用

islice関数を使用して、2Dリストで必要とされる特定の数の要素で特定のリストをスライスできます。したがって、ここでは2Dリストの各要素を調べ、その値を使用して元のリストを2スライスします。 islice関数を使用するにはitertoolsパッケージが必要です。

例

from itertools import islice
# Given list
listA = ['Sun','Mon','Tue','Wed','Thu','Fri','Sat']

# Length of 2D lists needed
len_2d = [ 3, 2]

# Use islice
def convert(listA, len_2d):
   res = iter(listA)
   return [list(islice(res,i)) for i in len_2d]
res = [convert(listA, len_2d)]
print("The new 2D lis is: \n",res)

上記のコードを実行すると、次の結果が得られます-

出力

The new 2D lis is:
[[['Sun', 'Mon', 'Tue'], ['Wed', 'Thu']]]