如何为Amazon Personalize构造批量推荐JSON输入数据

要生成Amazon Personalize批量推荐所需的每行一个JSON对象格式（JSON Lines），可以通过以下步骤实现：

1. 将userId转换为字符串类型：确保输出的JSON中userId值带引号，匹配目标格式要求。
2. 利用Pandas参数生成单行JSON对象：通过lines=True参数让每个JSON对象单独占一行，替代默认的数组包裹格式。

完整代码示例 #

import pandas as pd

# 创建DataFrame并将userId转为字符串类型
batch_prediction_user_id = pd.DataFrame({'userId': range(0,5)}).astype({'userId': str})
# 生成每行一个JSON对象的字符串
result = batch_prediction_user_id.to_json(orient="records", lines=True)

print(result)

输出结果 #

{"userId":"0"}
{"userId":"1"}
{"userId":"2"}
{"userId":"3"}
{"userId":"4"}

直接写入文件（可选） #

如果需要直接生成符合要求的批量输入文件，无需先生成字符串，可以直接指定文件路径：

batch_prediction_user_id.astype({'userId': str}).to_json('batch_input.json', orient="records", lines=True)

内容的提问来源于stack exchange，提问作者Learning