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