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

Question

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

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

不过，我还想快速检查一下这个小维度是否有效1.

有什么想法吗？

Answer 1

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

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

要快速检查最小尺寸是否为1，请使用：

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