pandas ランダムサンプリングの例でトランプのデッキを作る

import pandas as pd

# ハート、スペード、クラブ、ダイヤの頭文字
suits = ['H', 'S', 'C', 'D']

card_val = (list(range(1, 11)) + [10] * 3) * 4

base_names = ['A'] + list(range(2, 11)) + ['J', 'Q', 'K']
# ['A', 2, 3, 4, 5, 6, 7, 8, 9, 10, 'J', 'Q', 'K']

cards = []

for suit in ['H', 'S', 'C', 'D']:
    cards.extend(str(num) + suit for num in base_names)

deck = pd.Series(card_val, index=cards)

deck[:13]
# AH      1
# 2H      2
# 3H      3
# 4H      4
# 5H      5
# 6H      6
# 7H      7
# 8H      8
# 9H      9
# 10H    10
# JH     10
# QH     10
# KH     10
# dtype: int64

def draw(deck, n=5):
    return deck.sample(n)

draw(deck)
# 10C    10
# 9D      9
# 8S      8
# QC     10
# 5D      5
# dtype: int64

# カードの最後の文字を取得
get_suit = lambda card: card[-1]

deck.groupby(get_suit).apply(draw, n=2)
# C  QC    10
#    8C     8
# D  8D     8
#    5D     5
# H  4H     4
#    2H     2
# S  9S     9
#    KS    10
# dtype: int64