Python中的自定义长度矩阵

有时，当使用python创建矩阵时，我们可能需要控制给定元素在生成的矩阵中重复多少次。在本文中，我们将看到当元素以列表形式给出时，如何创建具有所需元素数量的矩阵。

使用压缩

我们声明一个包含要在矩阵中使用的元素的列表。然后，我们声明另一个列表，该列表将保存元素在矩阵中的出现次数。使用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']]

与映射和mul

在这种方法中，我们使用运算符模块中的mul方法代替了上面的zip方法。同样，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']