📝
[map,filterなど高階関数がよく分からない場合]listの中にlistを持っていた時にmap関数を使いたい
お詫び
Qiitaの元記事にて、区切り線を「---」で書いている場所があり、これがZennの記法に干渉して一部うまく表示できない記事がある事を認識しています。
全ての記事を精査しきれていないため、お手数ですがお見かけの際は教えていただけると大変喜びます。
[map,filterなど高階関数がよく分からない場合]listの中にlistを持っていた時にmap関数を使いたい
なんてことはないんだけど、解説してるサイトが見つからなかったので、後で困った時のためにメモ。
list_in_list = [
['1A','1B','1C'],
['2A','2B'],
['3A','3B','3C','4D']
]
list(
map(
print, list_in_list
)
)
#['1A','1B','1C']
#['2A','2B']
#['3A','3B','3C','4D']
list(map(lambda lists:list(map(print, lists)) , list_in_list))
#1A
#1B
#1C
#2A
#2B
#3A
#3B
#3C
#4D
リストの外の括弧[]に対してlist(map())でループを回しているイメージ。
発展
ただし、
list_in_list = [['first',['1A','1B','1C']],['second',['2A','2B']],['third',['3A','3B','3C','4D']]]
list(map(lambda lists:list(map(print, lists)) , list_in_list))
#first
#['1A','1B','1C']
#second
#['2A','2B']
#third
#['3A','3B','3C','4D']
のようなデータを扱うと、以下の場合おかしくなる。
list(map(lambda lists:list(map(lambda list2:list(map(print, list2)), lists)) , list_in_list))
#f
#i
#r
#s
#t
#1A
#1B
#1C
#s
#e
#c
#o
#n
#d
#2A
#2B
#t
#h
#i
#r
#d
#3A
#3B
#3C
#4D
これは、stringに'first'を入れてstring[0]、string[1]...のように取り出すような振る舞いがされている。
そのためエラーにならないが、期待値と異なる事がある。
解決
これを防ぎたい場合、たとえば今回の場合は以下のような書き方で逃げられる。
list(map(lambda lists:list(map(print, lists[1])) , list_in_list))
#1A
#1B
#1C
#2A
#2B
#3A
#3B
#3C
#4D
lists[0]のfirst, second, thirdを取りたい場合は、
list(map(lambda string: print(x[0]), list_in_list))
#first
#second
#third
で取れる。
落ち着いて一つずつ読み解けば難しい事ではないんだけど、勘違いしやすいので注意。
よく分からなくなった場合、list(map(..., list_in_list))をfor文に書き直してみると気付きやすいかも。
2018/04/09 解決追記
itertools.chainを使うともう少しスマートになる。
from itertools import chain
list_in_list = [['first', ['1A', '1B', '1C']], 'second', ['second',['2A', '2B']], ['third', [['3A', '3B', '3C', '4D']]]]
list(
map(print, list_in_list )
)
python2.x系の場合
printで説明しているのでprintについて言及するが、python2では上記をそのまま使えない。
そのため、以下のような関数を用意する。
def my_print(message):
print message
これでとりあえず使える。
読了後いいね！をお願いします。
どれだけの方に読んでもらっているか知りたいので、お手数をおかけしますがご協力いただけると嬉しいです。
Discussion