python - 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

if/else ステートメントを回避する方法があるかどうかを調べようとしています。何か案は？