[Python3]二次元配列における要素の計算方法について

Question

Python3です。
学習を初めて2,3日の初心者です。初投稿致します。

■ご質問したいこと
あるリスト
rects = [[2.1, 4.1],[0.5, 1.1],[2.5, 3.2]]
に対して、
各リスト内の積を算出し、最も大きい値のインデックスを返す、関数を定義したいです。

■現状の状況
rects = [[2.1, 4.1],[0.5, 1.1],[2.5, 3.2]]
import numpy as np
a = rects[0][0]*rects[0][1]
b = rects[1][0]*rects[1][1]
c = rects[2][0]*rects[2][1]
rects_cal = [a,b,c]
rests_cal_max = np.argmax(rects_cal)
print(rests_cal_max)

上記を実行すると、結果として「0」が表示され正しいのですが、
これを関数化するときの記載方法がわかりません、、。

■目的
def get_largest_rect(rects):
、、、、
return result

という表記でコードを作り、最後に、
get_largest_rect(rects=rects)で「0」を表示させたいです。

ご教示のほど、宜しくお願い致します。

Comments

[hayataka2049]

＞上記を実行すると、結果として「1」が表示され　0になるのですが

[csyu10064]

ご指摘ありがとうございます。0で正しく、修正致しました。

Answer 1

基本的な書き方としては、forループで掛け算した結果をlistに格納していき、できたlistに対してargmaxを呼べば良いでしょう。

def get_largest_rect(rects):
    lst = []
    for l in rects:
        lst.append(l[0]*l[1])
    return np.argmax(lst)

Comments

[csyu10064]

ご教示頂きありがとうございました。出来ました。
listとforの使い方、学習致します。引き続き宜しくお願い致します。

Answer 2

任意の要素数に対応させた関数を書いてみました。
numpyには多くの関数があるので、もっとすっきり書けるかもしれませんが。

rects = [[2.1, 4.1],[0.5, 1.1],[2.5, 3.2]]

def get_largest_rect(rects):
    import numpy as np
    ary = np.array(rects)
    ret = np.ones(ary.shape[0]).T
    for j in range(ary.shape[1]):
        ret *= ary[:,j]
    #print(ret)
    return np.argmax(ret)

ret = get_largest_rect(rects)
print(ret)

Comments

[csyu10064]

ご回答頂きありがとうございます。参考にさせて頂きます。