编写一个函数来从python中的矩阵列表中分离方阵？

Question

我正在尝试编写一个功能，如标题所说，但不断收到错误消息。这是我的代码：all_matrices = [A,B,C,D,E,F,G] #these are all matrices defined abovedef isSquare(matrices):     dims = []    square_matrices = []    i = 0    for i in range(len(matrices)):         dims.append(np.shape(matrices[i]))        if dims[i][0] == dims[i][1]:            square_matrices.append(matrices[i])       return square_matrices谢谢你们的第一个答案。现在我有一个新问题我不断得到'元组索引超出范围'但我不知道如何解决这个问题。有什么建议？顺便说一句，我正在使用 Spyder IDE。

Answer 1

由于您正在使用numpy，您可以像下面这样执行此操作

all_matrices = [np.zeros((3,3)), np.zeros((3,2)), np.zeros((4,4))]

square_matrices = [m for m in all_matrices if m.shape[0] == m.shape[1]]

row_matrices = [m for m in all_matrices if m.shape[0] < m.shape[1]]

col_matrices = [m for m in all_matrices if m.shape[0] > m.shape[1]]