コーディングテストで便利だと思った機能 (Python)

はじめに

僕はエンジニアを目指している大学生です。
コーディングテストで使って便利だなと思ったPythonの機能を紹介しようと思います。

便利な機能たち

リストをN個に分割

def split_list(l, n):
    for i in range(0, len(l), n):
        yield l[i:i + n]
 
l = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
result = list(split_list(l, 3))
print(result) # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

関数に yield を含めることで、ジェネレータになり反復可能になります。
また、ジェネレータを用いることで、都度必要な分だけ値を生成するので、メモリを効率的に使用できます。

累積和を生成

import itertools

l = [1, 2, 3, 4, 5, 6]

print(itertools.accumulate(l))
# <itertools.accumulate object at 0x10d648140>

print(list(itertools.accumulate(l)))
# [1, 3, 6, 10, 15, 21]

標準ライブラリである itertools の accumulate() を使うことで累積和を生成できます。
itertools.accmulate() の引数にはイテラブル(反復可能)なオブジェクトを指定します。
戻り値はイテレータなので、 for を回したり、 list() や tuple() を使って綺麗に表現できます。

文字列を逆順に出力

str = "abcdefg"
print(str[::-1]) # "gfedcba"

1つ目の引数(start)と2つ目の引数(end)に何も指定せず、3つ目の引数(step)に -1 を指定することで、末尾から1個ずつ遡って先頭まで要素を取得できます。

参照渡し回避

import copy

hoge = {1:{2:2, 3:3}}
print(id(hoge)) # 140418104920496

fuga = copy.deepcopy(hoge)
print(id(fuga)) # 140418107009728

fuga[1].update({4:4})

print(hoge == fuga) # False

print(hoge) # {1: {2: 2, 3: 3}}

print(fuga) # {1: {2: 2, 3: 3, 4: 4}}

copyモジュールの deepcopy() を使うことで、元のオブジェクトの完全な複製を作成することができるので、copy() と違い参照渡しを回避することが可能です。

重複をなくす

A = [1, 2, 3, 3, 4]
print(set(A)) # {1, 2, 3, 4}

set() を使うことで、指定したオブジェクトはハッシュテーブルになります。
ハッシュテーブルは重複要素を持てないため、重複をなくすことが可能です。
しかし、順序は保持されないです。

おわりに

今回はここまで紹介しました。
まだ紹介してない機能もあるので、次回紹介しようと思います。
エンジニア目指して頑張ります。