Pythonのカスタム長マトリックス

Pythonを使用してマトリックスを作成する場合、結果のマトリックスで特定の要素が繰り返される回数を制御する必要がある場合があります。この記事では、要素がリストとして指定されている場合に、必要な数の要素を含むマトリックスを作成する方法を説明します。

zipの使用

マトリックスで使用される要素を含むリストを宣言します。次に、マトリックス内の要素の出現回数を保持する別のリストを宣言します。 zip関数を使用して、要素を整理するためのforループを含む結果の行列を作成できます。

例

listA = ['m', 'n', 'p','q']

# Count of elements
elem_count = [1,0,3,2]

# Given Lists
print("Given List of elements: " ,listA)
print("Count of elements : ",elem_count)

# Creating Matrix
res = [[x] * y for x, y in zip(listA, elem_count)]

# Result
print("The new matrix is : " ,res)

出力

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

Given List of elements: ['m', 'n', 'p', 'q']
Count of elements : [1, 0, 3, 2]
The new matrix is : [['m'], [], ['p', 'p', 'p'], ['q', 'q']]

マップとマルチ付き

このアプローチでは、上記のzipメソッドの代わりに、オペレーターモジュールのmulメソッドを使用します。また、map関数はリスト内のすべての要素にmulメソッドを適用するため、forループは必要ありません。

例

from operator import mul

listA = ['m', 'n', 'p','q']

# Count of elements
elem_count = [1,0,3,2]

# Given Lists
print("Given List of elements: " ,listA)
print("Count of elements : ",elem_count)

# Creating Matrix
res = list(map(mul, listA, elem_count))

# Result
print("The new matrix is : " ,res)

出力

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

Given List of elements: ['m', 'n', 'p', 'q']
Count of elements : [1, 0, 3, 2]
The new matrix is : ['m', '', 'ppp', 'qq']