ndarrayで最後の行列の取得

Question

前提・実現したいこと

pythonのnumpyのndarrayの最後の要素の取得なのですが,行数が２以上の時は[-1]で最後の行を取得できるのですが、行数が１の時は[-1]で最後の要素しか取得できません。今、下に示すソースコードでaとbでどちらも[x,y,z]の形で取り出したいのですが、なにか良い方法はないでしょうか？

発生している問題・エラーメッセージ

該当のソースコード

import numpy as np
a=np.array([[1,2,3],[4,5,6]])
b=np.array([7,8,9])
print(anyfunction(a))#[4,5,6]とでてほしい
print(anyfunction(b))#[7,8,9]とでてほしい

"""
print(a[-1])->[4,5,6]
print(b[-1])->9
"""

試したこと

anyfunctionの中身を配列の大きさに着目してifで判断という手法を現在はとっています

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[namnium1125]

b=np.array([[7,8,9]])はダメということでしょうか？

[nono1234]

説明不足ですいません。namnium1125さんのおっしゃる通り、ｂをそのようにしたら[7,8,9]を返すのですが、その代わりaが[[1,2,3],[4,5,6]]になってしまうのです。aもbも入れたときにどちらも[x,y,z]の形で返すようなスマートなanyfunctionはないかということです。

[LouiS0616]

『その代わりaが[[1,2,3],[4,5,6]]になってしまうのです。』？？？現時点でその形式では？

[nono1234]

紛らわしい質問文で申し訳ありませんでした。ndimで次元数が取得できるのですね。今まではshapeやsizeで何とかしようとしていたのですが、私の本来のコードではaの行数が増えていくので途中でエラーを吐く可能性があり悩んでいました。解決できそうです。ありがとうございました。

Answer 1

aとbとで次元が異なりますから、二次元に揃えれば良いです。
具体的にはb = np.array([[7,8,9]])という風に。

ifで分岐するとしても、次元数で分岐するのが良いでしょう。

def anyfunction(arr):
    if arr.ndim == 1:
        return arr
    elif arr.ndim == 2:
        return arr[-1]
    raise ValueError

Comments

[nono1234]

申し訳ありません。ソースコードが長いので現象を示す簡単なコードを書いたのですが、問題の意図が伝わりにくくなってしまいました。
aが一次元でも二次元でも行を返すような書き方はないだろうかという質問です。おっしゃるとおりbは[7,8,9]を返しますが、その代わり同じ処理をaにもしなければならないのでaが[[1,2,3],[7,8,9]]を返すことになってしまいます。

[LouiS0616]

意味合い的には二次元に統一した方が良いと思いますが。

[nono1234]

まさにそのとおりです。そこから見直してみます。素早い回答ありがとうございました。

Answer 2

x,y,zが確定しているのなら。

def anyfunction(arr):
    return arr.reshape(-1, 3)[-1]

Comments

[nono1234]

reshapeの引数に-1を指定すると、いい感じに、しかも二次元配列に調整してくれるなんて初耳でした！！ありがとうございました！参考にさせていただきます