[Python] 如何用Python绘制小米新版天价logo

• ①利用 “超椭圆” 公式，绘制一个带填充色的椭圆；
• ② 遍历 “旧logo” 上的像素值，保留白色 “米字” 处的像素值；
• ③ 将带“米字” 的像素值，写入第一步的超椭圆中；
  & m, Z  Y2 _5 S* o  V6 B3 N" P! A- A

• ① plt.fill()：用于给图形填充颜色；
• ② plt.axis()：用于打开/取消坐标轴展示；
• ③ plt.subplots_adjust()和plt.margins()：用于去掉图形外侧多余的留白；
• ④ image.getpixel()：用于读取图片的像素值；
• ⑤ image.putpixel()：用于在指定坐标位置处，写入像素值；7 n. z! ]# D" _
  

复制代码
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image

复制代码
x = np.linspace(-1, 1, 1000)
y = + pow(1-abs(pow(x, 3)), 1/3) # 注意公式的写法
z = - pow(1-abs(pow(x, 3)), 1/3) # 注意公式的写法
plt.figure(figsize=(12, 12)) # 画布大小

plt.plot(x, y, color=(255/255, 103/255, 0))
plt.plot(x, z, color=(255/255, 103/255, 0))
plt.fill(x, y, color=(255/255, 103/255, 0)) # 填充颜色
plt.fill(x, z, color=(255/255, 103/255, 0)) # 填充颜色
plt.axis('off') # 不显示坐标轴
plt.subplots_adjust(top=1, bottom=0, right=1, left=0, hspace=0, wspace=0)
plt.margins(0, 0)
plt.savefig('fill.png', transparent=True, dpi=100) # 设置保存图片的透明值

复制代码
image_1 = Image.open('old.png')
image_2 = Image.open('fill.png')

w, h = image_1.size

for i in range(w):
    for j in range(h):
        data = image_1.getpixel((i, j))
        if data == (255,)*4:
            image_2.putpixel((i, j), (255,)*4)

image_2.save('new.png')