配列の要素の最小値を返す方法についての質問

Question

関数distance()と配列set=[set1,set2,set3,…]が既に定義されています。配列setの各要素のdistanceを計算し、その最小値を返すsetの要素を調べたいです。つまり、distance(set1),distance(set2),distance(set3)…とみていったときに、distanceを最小にする、setXを知りたいということです。

私は次のようにコーディングしましたが、もっと良い方法はありますか？

minimam = 100000
for p in set
    if minimam > distance(p):
       minimam = distance(p)
       m = p

こうすれば、mの最終値が欲しかったものになります。もっとスマートに（例えば１行で）ソースをかけないものでしょうか。

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

編集**
distanceが引数を二つとる関数で、一方を固定している場合、どうしたらよいでっしょうか。つまり、

minimam = 100000
j = 1
for p in set
    if minimam > distance(p,j):
       minimam = distance(p,j)
       m = p

を一行で書くにはどうしたらよいのでしょうか。

Answer 1

minimum = min(set, key=distance)

Pythonのset型の型名を上書きしてしまうので、変数名にsetは使わない方がいいですよ。

---

1変数関数でなければならないので1変数関数を定義するしかないです。

j = 1

def distance_for_min(x)
    return distance(p,j)

minimum = min(set, key=distance_for_min)

とかですね。ローカル変数を束縛した関数を定義します。

あるいはラムダで

j = 1
minimum = min(set, key=lambda x: distance(x, j))

と書きます。

Comments

[physics303]

ありがとうございます。できれば、何をしているのかを教えてもらえないでしょうか。keyで調べてみたのですが、いまいちわからず。

それから、distanceが多変数の場合はどうでしょうか。つまり、

minimam = 100000
j = 1
for p in set
if minimam > distance(p,j):
minimam = distance(p,j)
m = p
のような場合にmを一行で求めるにはどうしたらよいのでしょうか。

[quickquip]

質問のソースと同じことをするわけなので、何をしているのかを解説する意味はありません。わからないのは「関数そのものを引数で渡す」という概念でしょうか?