如何用Python绘制Google QuickDraw数据集?

2024-04-20 16:10:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经从QuickDraw数据集(JSON格式)提取了笔划数据,但是我不知道如何用Python绘制它。我试了好几次,但都没用。我需要关于用Python绘制笔划的帮助。以下是我提取笔划数据的代码:

import json
file = open(filepath)
lines = file.readlines()

data = json.loads(lines[0])
strokes = data['drawing']
x = strokes[0][0]
y = strokes[0][1]

以下文本是数据文件的第一行:

{"word":"airplane","countrycode":"US","timestamp":"2017-03-08 21:12:07.26604      UTC","recognized":true,"key_id":"5152802093400064","drawing":[[[167,109,80,69,58,31,57,117,99,52,30,6,1,2,66,98,253,254,246,182,165],[140,194,227,232,229,229,206,124,123,149,157,159,153,110,82,77,74,109,121,127,120]],[[207,207,210,221,238],[74,103,114,128,135]],[[119,107,76,70,49,39,60,93],[72,41,3,0,1,5,38,70]]]}

Tags: 数据代码importjsondata格式绘制open
2条回答

除了Swati的答案之外,还有一个^{} python package使这变得简单:

from quickdraw import QuickDrawData

qd = QuickDrawData()

anvil = qd.get_drawing("anvil")
anvil.image.show()

如果要手动执行此操作,可以zip()读取所有的x值和y值,并进行渲染(使用PIL或其他任何方法):

import json
from PIL import Image, ImageDraw

# read ndjson lines
lines = open('full_simplified_cat.ndjson','r').readlines()
# grab the first line, JSON parse it and fetch the 'drawing' array
raw_drawing = json.loads(lines[0])['drawing']

print('before',raw_drawing)
# zip x,y coordinates for each point in every polyline
polylines = (zip(polyline[0], polyline[1]) for polyline in raw_drawing)
# notice how the data is shuffled to (x1,y1),(x2,y2) order
print('after',polylines)

# make a new image
pil_img = Image.new("RGB", (240, 270), (255,255,255))
# get a drawing context
d = ImageDraw.Draw(pil_img)
# render each polyline
for polyline in polylines:
    d.line(polyline,fill=(0, 0, 0),width=3)
# display image
pil_img.show()

相关问题 更多 >