Python實現圖片轉字符畫

主要學習了PIL，argparse，list創建，open等函數的使用。

# -*- coding=utf-8 -*-
#PIL python imaging library, pillow是PIL的一個分支，
#如今已經發展成爲比PIL本身更具活力的圖像處理庫
from PIL import Image
import argparse
#argparse 是python自帶的命令行參數解析包，可以用來方便地讀取命令行參數。它的使用也比較簡單。https://blog.csdn.net/yy_diego/article/details/82851661

#命令行輸入參數處理
parser = argparse.ArgumentParser()

parser.add_argument('file')     #輸入文件
parser.add_argument('-o', '--output')   #輸出文件
parser.add_argument('--width', type = int, default = 80) #輸出字符畫寬
parser.add_argument('--height', type = int, default = 80) #輸出字符畫高

#獲取參數
args = parser.parse_args()

IMG = args.file
WIDTH = args.width
HEIGHT = args.height
OUTPUT = args.output

#Python創建列表的四種方式https://blog.csdn.net/u014612521/article/details/100159907
ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. ")

# 將256灰度映射到70個字符上
def get_char(r,g,b,alpha = 256):
    if alpha == 0:
        return ' '
    length = len(ascii_char)
    gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)

    unit = (256.0 + 1)/length
    return ascii_char[int(gray/unit)]

if __name__ == '__main__':

	#open的用法https://blog.csdn.net/NeverLate_gogogo/article/details/85292663
    im = Image.open(IMG)
    im = im.resize((WIDTH,HEIGHT), Image.NEAREST)

    txt = ""

    for i in range(HEIGHT):
        for j in range(WIDTH):
            txt += get_char(*im.getpixel((j,i)))
        txt += '\n'

    print(txt)

    #字符畫輸出到文件
    if OUTPUT:
        with open(OUTPUT,'w') as f:
            f.write(txt)
    else:
        with open("output.txt",'w') as f:
            f.write(txt)