はむこの勉強記録

http://bit.ly/2ktf20t の写し

AtCoder Beginner Contest 056 - C - Go Home (python)

abc056.contest.atcoder.jp

np.whereはなぜかtupleが帰ってくる。()はtuple, []はlist, {}はdict or set。tupleは[0]などでアクセス可能。np.where(a>1)は(a>1).nonzero()でも同様にインデックスが取得できるらしい。

stackoverflow.com