python - 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