◆CheckiO
というサイトの練習問題にあった課題。
記事の題名通りどおり、配列の値がすべて等しいか調べる、という内容である。
初心者の自分は、以下のような解答をした。
python.all_the_same.py
from typing import List, Any
def all_the_same(elements):
#配列の要素数が0の場合
if len(elements)==0:
return True
#配列の先頭要素を取得
lead_element=elements[0]
#要素が等しくない場合Falseを返す
for element in elements:
if lead_element!=element:
return False
return True
この関数で要件は満たしてはいるのだが、
もっと簡潔なコードはないかと色々調べてみた。
このような時、CheckiO は、
全世界のプログラマの解答が閲覧できるため、便利だ。
備忘録がてら、勉強になったモノをいくつか紹介したいと思う。
1.スライスを使う
python.all_the_same.py
def all_the_same(elements):
return elements[1:] == elements[:-1]
いちばん驚いた。
list=[1,2,1]
のようなリストを引数として与えた場合
左辺:[2,1]
右辺:[1,2]
で比較し、論理値を返すようになっている。
もちろん同じではないのでFalseが返ってくる。
上記の例だけでは分かり辛いので補足する。
◆計算過程のprintを追加したコード
python.all_the_same.py
def all_the_same(elements):
#計算過程を出力しておく
print(str(elements[1:])+"=="+str(elements[:-1]))
return elements[1:] == elements[:-1]
print(all_the_same([1,2,3,4]))
◆結果
[2,3,4]==[1,2,3]
False
当然ながら結果はFalseだが、
上記のようなスライス記法で、
「配列二つ目から末尾まで」
と
「配列先頭から末尾一つ手前まで」
を比較できる事がわかる。
スゲェよ旦那ぁ！
2.setとlenを使う
python.all_the_same.py
def all_the_same(elements):
return len(set(elements)) <= 1
◆setとは
・集合を表すデータ型
・設定時に重複が取り除かれる
・index番号を持たない
・s=set([1,2,3])みたいな感じで作成可能
◆lenとは
・len(変数名 or 文字列)で長さを返す
setの特徴のひとつ、
「重複が取り除かれる」という部分を利用し、
要素数が1以上だった際にFalseを返している。
覚えておくと役に立ちそう。