如果你有一维和二维数组，如何获得行数？

Question

我有两个数组，如下所示，想通过函数获取行数。.shape.

X = np.array([0, 4, 3, 5, 1, 2])

Y = np.array([[-1, 0, 4, 4],
              [ 1, 0, 5, 0],
              [ 2, 7, 4, 0],
              [ 3, 0, 4, 9],
              [ 4, 6, 4, 0]])

X.shape[0]
Y.shape[0] 

结果是

6
5

因为 X 是一个有1行的矩阵，我想 X.shape[0] 返回 1. 然而，它返回 6 即列数。请你推荐一个函数来实现我的目标？

Answer 1

从@Divakar的评论来看，实现这个目标的命令是 np.atleast_2d.