Python学习

2021-03-12

字数统计: 722字 | 阅读时长≈ 3分

Python小程序

最近学习python，打算写一些小程序，放到这里总结一下。

Python实现把图像转换为字符画

'''
pip3 install --upgrade pip
pip3 install pillow
灰度值公式 gray ＝ 0.2126 * r + 0.7152 * g + 0.0722 * b
'''
from PIL import Image
import argparse # argparse库是用来管理命令行参数输入的

# 首先，构造命令行输入参数处理 ArgumentParser 实例
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

# 字符集
ascii_char = list('"$_&WM#*oahkbdpqwmZO0QLCJUYXzcvunxr\jft/\|()1{}[]?-/+@<>i!:;,\^`.')

# RGB 值转字符的函数,alpha 值为 0 的时候表示图片中该位置为空白
def get_char(r,g,b,alpha = 256):
    # 判断 alpha 值
    if alpha == 0:
        return ' '

    # 获取字符集的长度
    length = len(ascii_char)

    # 将 RBG值转换为灰度值 gray，灰度值范围为 0~255
    gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)

    # 灰度值范围为 0-255，而字符集只有 70
    # 需要进行如下处理才能将灰度值映射到指定的字符上
    unit = (256.0+1)/length

    # 返回灰度值对应的字符
    return ascii_char[int(gray/unit)]

if __name__ == '__main__':      #表示如果 ascii.py 被当作 python 模块 import 的时候，这部分代码不会被执行
    # 打开并调整图片的宽和高
    im = Image.open(IMG)
    im = im.resize((WIDTH,HEIGHT), Image.NEAREST)

    # 初始化输出的字符串
    txt = ""
    
    # 遍历图片中的每一行
    for i in range(HEIGHT):
        # 遍历该行中的每一列
        for j in range(WIDTH):
            # 将 (j,i) 坐标的 RGB 像素转为字符后添加到 txt 字符串
            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)

注意：

1、关于PIL库的问题，使用pip安装后仍有可能无法找到Image，原因是在python3.4之后就不支持Image了，所以这样使用

from PIL import Image, ImageTk

PIL库是一个第三方 Python 图像处理库，argparse 库是用来管理命令行参数输入的一个 python 内置库。　　

2、灰度值

灰度值：指黑白图像中点的颜色深度，范围一般从 0 到 255，白色为255，黑色为 0，故黑白图片也称灰度图像
我们可以使用灰度值公式将像素的 RGB 值映射到灰度值：gray ＝ 0.2126 * r + 0.7152 * g + 0.0722 * b

本文作者： y0lo
本文链接： http://example.com/2021/03/12/Python学习/
版权声明： 本博客所有文章除特别声明外，均采用 MIT 许可协议。转载请注明出处！