如何使用 DataFrame 使用 Jinja 有条件地创建文档？

Question

对于 DataFrame 中的每一行，我想为 avro 模式打印一些值。我有一个看起来像这样的 DataFrame。

import pandas as pd

x = pd.DataFrame({'column' : ['id','type','time']
    , 'desc': ['the sk',' the type' , 'epoch time']
    ,'nullable':[False, False, True]})

print x

对于 DataFrame 中的每一行，我想为 Jinja 模板创建输入

所以我的输出看起来像：

{
name       : "id"
description: "the sk"
}
{
name       :"type"
description:"the type"
}
{
name       :"time","null"
description:"epoch time"
}

我该怎么做？我还想根据数据框中的其他列值有条件地嵌套结果。

Answer 1

我认为将DataFrame 转换为dict 是解决方案。我使用orient='records' 来获得适合渲染的dict。

from jinja2 import Template

# Template definition
template = Template("""{% for row in data %}
{
name:  {{ row['column'] }}{{ ', null' if row['nullable'] }}
description: {{ row['desc'] }}
}
{% endfor %}""")

# Rendering
print(template.render(name='John Doe', data=x.to_dict(orient='records')))

# {
# name:  id
# description: the sk
# }
# 
# {
# name:  type
# description:  the type
# }
#
# {
# name:  time, null
# description: epoch time
# }