使用python自動化png格式

Question

我試圖找出一種方法來自動化png格式以添加標題字幕和帶有徽標圖像和源代碼的頁腳欄。我想用python來完成這個圖像格式，因爲我對這種語言最熟悉。我正在尋找一些方向來說明哪些模塊可以很好地用於這樣的事情？理想情況下，該腳本的用戶對此過程看起來像這樣。

1）用戶將有一個看起來是這樣的一個PNG圖像：

2）用戶將啓動腳本：

python autochart_formatting.py 

3）腳本會提示用戶輸入以下信息：

• 輸入圖表標題：
• 輸入圖表副標題：
• 輸入源：
• 調查名稱：
• 樣本n =：
• 輸入路徑，以圖表爲png你想要的格式：
• 輸入要格式化的路徑圖像保存：

3）使用該信息PNG將與被格式化爲看起來像這樣：

Answer 1

Pillow (the maintained successor to PIL: Python Imaging Library)可以處理你想要的。

您可以在檢索用戶輸入後擴展圖像並放置文本。下面是添加標題的示例：

from PIL import Image, ImageFont, ImageDraw 

img = Image.open('my_chart.png') 
w,h= img.size 

# put pixels into 2D array for ease of use 
data = list(img.getdata()) 
xy_data = [] 
for y in xrange(h): 
    temp = [] 
    for x in xrange(w): 
     temp.append(data[y*w + x]) 
    xy_data.append(temp) 

# get the title 
title = raw_input("Title:") 

# load the font 
font_size = 20 
font = ImageFont.truetype("/path/to/font.ttf",font_size) 

# Get the required height for you images 
height_needed = font.getsize(title)[1] + 2 # 2 px for padding 

# get the upperleft pixel to match color 
bg = xy_data[0][0] 

# add rows to the data to prepare for the text 
xy_data = [[bg]*w for i in range(height_needed+5)] + xy_data # +5 for more padding 

# resize image 
img = img.resize((w,h+height_needed+5)) 

# convert data back to 1D array 
data = [] 
for line in xy_data: 
    data += line 

# put the image back in the data 
img.putdata(data) 

# get the ImageDraw item for this image 
draw = ImageDraw.Draw(img) 

# draw the text 
draw.text((5,0),title,font=font,fill=(0,0,0)) # fill is black 

img.save('titled_plot.png') 

Answer 2

這完全在Pillow（仍在維護的Python成像庫的一個分支）的能力範圍內。

如果你不想推出你自己的圖形代碼，你也可以用matplotlib來做到這一點。它會讓你在圖形格式化方面略微缺乏靈活性，但創建速度會更快。