如何在if语句中使用数据类型

Question

我正在编写一个程序来检查矩阵是否为正方形（行和列的数量等于2x2、3x3等）

我认为最好使用内置的size函数计算元素并取平方根。我想编写一个if语句，如果平方根不等于整数，它会打印一条错误语句，但是我不确定如何在我的语句中指定整数。

这是我尝试过的

    import numpy as np
    A = np.array([[1,2,3],[4,5,6]])
    check = A.size
    if check**.5 ...

我不知道在接下来的声明中要写些什么

Answer 1

您可能要检查shape属性。

if A.shape[0] == A.shape[1]:
    # Is square
else:
    # Is not square

Answer 2

您可以检查它是否为整数。

if check**.5 == int(check**.5):
    ...