最小値を持つ行の位置番号の取得

Question

実現したいこと

ランダムに生成された配列(a1)の最小値を持つ要素の各行の位置番号を取得し、変数(res)に代入したいです。
どのような関数などを用いて実装すればよいかが分かりません。

よろしくお願いいたします。

ソースコード

import numpy as np

gen = np.random.RandomState(0)
a1 = gen.randn(8, 8)

res = 
print(res)

調べたこと

argminやaxis=1などを使うのかもしれませんが、よく分かりませんでした。

###追記

前提が不足していました。すみません。

私は、ネット上でPythonを学べるサイト(https://signate.jp/)でNumPyについて学んでいます。
問4として、
「a1の最小の値を持つ要素の行の位置番号をresに代入してください。」
という課題が出ました。

しかし、実際にres = a1.argmin(axis=1)を試しましたが、
「変数resに意図した値が代入されていません。」
という評価で返ってきます。

他の方がおっしゃっていた、下記のコードについても同じ評価でした。

res = np.unravel_index(np.argmin(a1), a1.shape)
print(res)

他に何かやり方があるのでしょうか。
よろしくお願いいたします。

Comments

[LouiS0616]

> argminやaxis=1などを使うのかもしれませんが

試してみましたか？

[Hiro-Python]

質問内容を補填しました。

Answer 1

ご自分ですでに見つけておられるargminにaxis=1を渡せばいいだけではないでしょうか。

In [1]: import numpy as np
   ...:
   ...: gen = np.random.RandomState(0)
   ...: a1 = gen.randn(8, 8)
   ...:
In [2]: a1.argmin(axis=1)
Out[2]: array([5, 0, 4, 1, 1, 2, 0, 7])

Answer 2

こうかな？

import numpy as np

gen = np.random.RandomState(0)
a1 = gen.randn(8, 8)
print(a1)

res = np.unravel_index(np.argmin(a1), a1.shape)
print(res)
"""
(2, 4)
"""
print(a1[res[0],res[1]])
"""
-2.5529898158340787
"""