python一鍵將白色背景變為透明背景

代碼:

需要pillow庫
安裝方法：pip pillow

• 有一說一，比ps好用多了

from PIL import Image
pic = Image.open(r'C:\Users\xxxx\Desktop\origin.png')
pic = pic.convert('RGBA') # 轉為RGBA模式
width,height = pic.size
array = pic.load() # 獲取圖片像素操作入口
for i in range(width):
    for j in range(height):
        pos = array[i,j] # 獲得某個像素點，格式為(R,G,B,A)元組
        # 如果R G B三者都大於240(很接近白色了，數值可調整)
        isEdit = (sum([1 for x in pos[0:3] if x > 240]) == 3)
        if isEdit:
            # 更改為透明
            array[i,j] = (255,255,255,0)

# 保存圖片
pic.save('result.png')