使えるかもしれないpyhtonの小さな知識ポイント

スワップ値

>>> a = 1
>>> b = 2
>>> a , b = b , a
>>> a
2
>>> b
1

リストのアンパッケージ

>>> a, b, c = [1, 2, 3]
>>> a
1
>>> b 
2
>>> c
3
>>> a, *other = [1, 2, 3]
>>> a
1
>>> other
[2, 3]

拡張リスト

>>> i = ['a', 'b', 'c']
>>> i.extend(['e', 'f', 'g'])
>>> i
['a', 'b', 'c', 'e', 'f', 'g']

リストの負のインデックス

>>> a = [1,2,3]
>>> a[-1]
3

リストスライス

>>> a = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> a[3:6] #  3   6      
[3, 4, 5]
>>> a[:5] #  5   
[0, 1, 2, 3, 4]
>>> a[5:] #  5   
[5, 6, 7, 8, 9]
>>> a[::] #     （    ）
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> a[::2] #    
[0, 2, 4, 6, 8]
>>> a[1::2] #    
[1, 3, 5, 7, 9]
>>> a[::-1] #     
[9, 8, 7, 6, 5, 4, 3, 2, 1, 0]

2 D配列へんい1 D配列

import itertools
>>> a = [[1, 2], [3, 4], [5, 6]]
>>> i = itertools.chain(*a)
>>> list(i)
[1, 2, 3, 4, 5, 6]

インデックス付き反復

>>> a = ['Merry','Christmas','Day']
>>> for i, x in enumerate(a):
 ...    print('{}: {}'.format(i,x))
 ...
 0: Merry
 1: Christmas
 2: Day

リスト導出式

>>> le = [x * 2 for x in range(10)]
>>> le #     2 
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

>>> le = [x for x in range(10) if x % 2 == 0]
>>> le #      
[0, 2, 4, 6, 8]

ジェネレータ式

>>> ge = (x * 2 for x in range(10))
>>> ge
 at 0x01948A50>
>>> next(ge)
0
>>> next(ge)
4
...
>>> next(ge)
Traceback(most recent call last):
  File "",line 1, in 
StopInteration

しゅうごうゆうどうしき

>>> nums = {n ** 2 for n in range(10)}
>>> nums
{0, 1, 64, 4, 36, 9, 16, 49, 81, 25}

ディクショナリガイド

>>> d = {s : len(s) for s in ["one", "two", "three"]}
>>> d
{'one': 3, 'two': 3, 'three': 5}

2つの辞書をマージ

>>> def merge_dicts(d1,d2):
        return {k: v for d in (d1,d2) for k, v in d.items()}
>>> merge_dicts({"1":"2"}, {"a":"b"})
{'1':'2', 'a':'b'}

文字列の結合

>>> i = ['a', 'b', 'c']
>>> "".join(i)
'abc'

keyが辞書に存在するかどうかを判断する

>>> d = {"1":"a"}
>>> d['2']
Traceback (most recent call last):
  File "", line 1, in 
KeyError: '2'
>>> '1' in d
True
>>> d['1']
'a'
>>> d.get("1")
'a'
>>> d.get("2")
>>>

さんげんえんざん

>>> d = {'1','a'}
>>> exist = True if '2' in d else False
>>> exist
False

デコレーション

from functools import wraps 

def tags(tag_name):
  def tags_decorator(func):
    @wraps(func)
    def func_wrapper(name):
      return "{1}{0}>".format(tag_name,func(naem))
    return func_wrapper
  
@tags("p")
def get_text(name):
  """returns some text"""
  return "Hello " + name

print(get_text("Python"))
>>> 
Hello Python

ディクショナリのサブセット

>>> def sub_dicts(d, keys):
 ...    return {k:v for k, v in d.items() if k in keys}
 ...
>>> sub_dicts({1:'a', 2:'b', 3:'c'},[1, 2])
{1:'a', 2:'b'}

2つのリスト変換辞書

>>> a = [1, 2, 3]
>>> b = ['a', 'b', 'c']
>>> dict(zip(a,b))
{1:'a', 2:'b', 3:'c'}

辞書を反転

>>> d = {'a':1, 'b':2, 'c':3, 'b':4}
>>> zip(d.values(), d.keys())

>>> z = zip(d.values(), d.keys())
>>> dict(z)
{1:'a', 2:'b', 3:'c', 4:'d'}

具名タプル

>>> from collectons import namedtuple
>>> Point = namedtuple("Point", "x,y")
>>> p = Point(x = 1, y =2)
>>> p.x
1
>>> p[0]
1
>>> p.y
2
>>> p[1]
2

デフォルト辞書

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> d["a"]
[]
>>> d["a"].append(1)
>>> d["a"]
[1]

>>> d = defaultdict(lambda: [1])
>>> d
defaultdict( at 0x00c0DB28>, {})
>>> d['a']
[1]

辞書のデフォルト値の設定

>>> d = dict()
>>> if 'a' not in d:
 ...    d['a'] = []
 ...
>>> d['a'].append(1)
>>> d
{'a': [1]}

>>> d.setdefault('b',[]).append(2)
>>> d
{'a':[1], 'b':[2]}

しゅうはすうとうけい

>>> from collections import Counter
>>> import random
>>> a = [random.randint(0,10) for __ in range(20)]
>>> a
[5, 4, 9, 6, 4, 2, 4, 6, 9, 5, 5, 1, 3, 6, 9, 1, 6, 1, 5, 5]
>>> Counter(a)
Counter({5: 5, 6: 4, 1: 3, 4: 3, 9: 3, 2: 1, 3: 1})

ちくじじしょ

>>> d = dict((str(x), x) for x in range(10))
>>> d.keys() # key   
dict_keys(['8', '2', '6', '9', '1', '0', '5', '4', '7', '3'])

>>> from collections import OrderedDict
>>> m  = OrderedDict((str(x), x) for x in range(10))
>>> m.keys() # key          
odict_keys(['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'])

リスト内の最大最小の最初のn個の数

>>> import heapq
a = [51, 95, 14, 65, 86, 35, 85, 32, 8, 98]
>>> heapq.nlargest(5,a)
[98, 95, 86, 85, 65]
>>> heapq.nsmallest(5, a)
[8, 14, 32, 35, 51]

for ... else

found = False
for i in foo:
    if i == 0:
        found = True
        break
if not found:
    print("i was never 0")
    

for i in foo:
    if i == 0:
        break
else:
    print("i was never 0")

ファイルを開く

>>> with open('foo.txt', 'w') as f:
...     f.write("hello")