Я ищу аккуратный способ вернуть несколько матриц в python

Question

Я пытаюсь найти более аккуратный способ возврата произвольного числа NxN-матриц. Сейчас я использую эту функцию

from numpy import matrix, zeros 

def empty_matrix(dim, num): 
"""Returns an empty square matrix of type complex and size a.""" 
    if num == 1: 
     return matrix(zeros(shape =(dim, dim), dtype=complex)) 
    else: 
     return [ matrix(zeros(shape =(dim, dim), dtype=complex)) for _ in range(num)] 

следующим образом:

A,B,C = empty_matrix(2, 3) # sets A, B, C as 3 2x2 matrices 

Я пытаюсь выяснить, если есть способ избежать, если/другое заявление. Есть идеи?

Answer 1

Отбросьте if num == 1, а также использовать 1-элементный кортеж-распаковку:

A,B,C = empty_matrices(2, 3) # sets A, B, C as 3 2x2 matrices 
A, = empty_matrices(2, 1) # sets A as a 2x2 matrix