Python列表理解列表-获取最大值的好方法

我正在写一个函数，它应该首先找出输入向量所在空间的维数。向量的方向未知，因此可以是1*ndim或ndim*1。我想找一个漂亮的一班轮这样做。我试过这个，但这不是正确的语法：

 ndim = [max(n,p) for n,p in (n,p) = np.shape(X) if min(n,p) = 1]

顺便说一句，我还想快速检查小尺寸是否有效为1

---

有什么想法吗？

我认为以下内容修复了您表达式中的错误：

ndim = [max(n,p) for n,p in [np.shape(X)] if min(n,p) == 1]

要快速检查最小尺寸是否为

1

，请使用：

min(*np.shape(x)) == 1

怎么样

assert min（X.shape）==1；ndim=最大（X形）