Pythonにもmap関数があり、容易に配列から新たな配列を生み出すことができます。lambdaを用いて無名関数を使うなど、いくつか応用方があるので備忘録がてらまとめていこうと思います。

簡単のため、ある配列の全ての要素の絶対値をとるスクリプトで試します。

mapと無名関数

新たに配列から配列を作るために忘れてはいけないのは、mapオブジェクトをlistで囲むことです。

その点を考慮しつつ、新たな配列を生み出していきます。まずは、無名関数lambdaを使います。

a = [-3, -2, -1, 0, 1, 2, 3]
to_abs = lambda x: abs(x)

b = list(map(to_abs, a))
print(b)

[3, 2, 1, 0, 1, 2, 3]

無名関数to_absにより、渡される各関数の値の絶対値を返します。

mapと普段使いの関数

普段通りの関数宣言でも動きます。

def to_abs(x):
    return abs(x)
    
c = list(map(to_abs, a))
print(c)

[3, 2, 1, 0, 1, 2, 3]

普通に動きますが、ここで注意したいのはmapに渡す関数に直接的に引数を与えないことです。

そうすることで難なく動きます。

引数を渡してみるとどうなるでしょうか。

def to_abs(x):
    return abs(x)
    
c = list(map(to_abs(x), a))
print(c)

Traceback (most recent call last):
  File "map.py", line 3, in <module>
    c = list(map(to_abs(x), a))
NameError: name 'x' is not defined

参照するxがないのでエラーが発生します。その点だけ注意しましょう。

setについても同様に動作します

d = set(map(to_abs, a))
print(d)

{0, 1, 2, 3}

しっかりと集合として捉えられていますね。

mapを使ってさせることはそこまで難しくないですが、処理速度的にはおそらく内包表記の方が速いと思われるので場合に合わせて導入するのが良いかもしれません。

ただ、numpyやpandasでデータ分析をする際は、map関数がそれぞれで与えられていることもあるので、使えるに越したことはないでしょう。