二次元配列で、２番目の要素が最大のものをピックアップしたい

Question

二次元配列で、２番目の要素が最大のものをピックアップしたいです。
select_one配列に[[1, 3, 1], [1, 6, 6]] が入っています。
select_one[0][1]は３・select_one[1][1]は６ですが
この配列の２番目の要素が最大の配列を取り出したいです。

select_one=[[1, 3, 1], [1, 6, 6]] 

のときは

[1, 6, 6]

が返ってこさせたいです。
ただこのselect_one配列は要素数が２つと限定されているわけではなく、

select_one=[[1, 3, 1], [1, 6, 6], [1, 7, 5],[1,2,2],[1,3,3]] 

など２つ以上の要素を持つ可能性もあり、また

select_one=[[1, 3, 1]] 

と要素を１つのみもつ場合もあります。（もちろん要素が一つのこの場合は、[1, 3, 1]が返ってくることを期待します）

今、

for i in range(len(select_one)):
     if select_one[i][1] 

とコードを書いたのですが、大小比較できるその先がどうしても思いつきません。
なんてコードを書けばよいのでしょうか？

Answer 1

def max_of_middle(a):
  return a[1]

select_one=[[1, 3, 1], [1, 6, 6], [1, 7, 5],[1,2,2],[1,3,3]] 
print(max(select_one, key = max_of_middle))

max 関数の key パラメータに関数を渡し、そこで2番目の要素を返してあげれば良いかと思います。