python 圖像處理，畫一個正弦函數

import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
import math

size = 300
new_im = Image.new("RGBA",(size,size)) #創建一個空的圖片

a_img = np.array(new_im)  #獲取空圖片的數據

m = (size-50)/2 #函數的擺幅
for i in range(0,size):
    x = 2 * math.pi * (  i / size )  #控制 根據 i / size 的比例，控制 x 的值為  0-2π
    y = size/2 - m*math.sin(x)  #將正弦函數畫在中間
    y = int(y)
    if y < size and y >= 0:
        a_img[y][i] = (0,0,0,255)   


plt.figure("beauty")
plt.imshow(a_img)
#plt.axis('off')
plt.show()

　　

效果為：