【Python】あみだくじ画像を自動生成する方法！候補者指定するだけ！

こんにちは、ヒガシです。

このページではPythonを使って以下の画像のようなあみだくじを一瞬で自動生成する方法をご紹介していきます。

あなたがやることは候補者のリストを作成するだけでOKです。

それではさっそくやっていきましょう！

必要なライブラリ一覧
あみだくじ自動作成プログラムの紹介
サンプルコードの実行結果確認
おわりに

必要なライブラリ一覧

今回は以下のライブラリを使用します。

★Pillow（要インストール）

★numpy（要インストール）

まずはあなたの環境でインポート可能か確認してみましょう。

あみだくじ自動作成プログラムの紹介

さっそくですが、冒頭に紹介したようなあみだくじを自動で生成できるプログラムをご紹介します。

#ライブラリインポート
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
import numpy as np

#情報入力
list=['AA','BB','CC','DD','EE','FF','GG','HH']
bar_horizontal=30
#文字サイズ、色、画像サイズ等の指定
font_size=50
font_color=255
line_w=5
h=1200
w=1200
font_path="C:\Windows\Fonts\HGRPP1.TTC"
font = ImageFont.truetype(font_path, font_size) 
bar_vertical=len(list)
width_h=int((h-400)/bar_horizontal)
width_w=int(w/bar_vertical)

#ベース画像の作成
base_img=Image.new("L",(w+width_w,h+width_h))
#PillowからNumPyへの変換
base_img=np.array(base_img)

#あみだの線を描写する
for i in range(bar_vertical):
    base_img[width_h+100:-width_h-100,(i+1)*width_w-line_w:(i+1)*width_w+line_w]=200
for i in range(bar_horizontal):
    pos=np.random.randint(1,bar_vertical)
    base_img[width_h+200+int((i+1)*width_h-line_w):width_h+200+int((i+1)*width_h+line_w),pos*width_w:(pos+1)*width_w]=200

#NumPyからPillowへの変換
base_img=Image.fromarray(base_img)
#候補初リストを描写する
for i in range(bar_vertical):
    ImageDraw.Draw(base_img).text(((i+1)*width_w-12*len(list[i]), 50), list[i], font = font , fill = font_color)
election=np.random.randint(0,bar_vertical)
ImageDraw.Draw(base_img).text(((election+1)*width_w-25, h-100), '★', font = font , fill = font_color)

#画像の表示＆保存
base_img.show()
base_img.save('amida3.jpg', quality=100)

基本的にあなたがやるべきことは、８行目の候補者のリストを作成するだけでOKです。

もし候補者が非常に多く、横棒がスカスカになった場合は９行目のbar_horizontalの数値を増加させましょう。ここで横棒の数を指定しています。

また、同じく候補者が多く、縦棒の間隔が狭くなりすぎた場合は１５行目のwを増加させましょう。ここで生成画像の横幅を指定しています。