我想以与现有txt文件匹配的方式保存我的数据帧(我有一个基于此txt文件的训练模型,现在我想预测需要匹配此格式的新数据).
目标TXT文件如下所示(前3行):
2 qid:0 0:0.4967141530112327 1:-0.1382643011711847 2:0.6476885381006925 3:1.523029856408025 4:-0.234153374723336
1 qid:2 0:1.465648768921554 1:-0.2257763004865357 2:0.06752820468792384 3:-1.424748186213457 4:-0.5443827245251827
2 qid:0 0:0.7384665799954104 1:0.1713682811899705 2:-0.1156482823882405 3:-0.3011036955892888 4:-1.478521990367427
第一列只是一个随机整数(这里是2和1) Qid总是通过冒号连接到一个整数. 然后,对于其余的列,整数后面跟着一个浮点数.
我的数据帧如下所示:
data = {'label': [2,3,2],
'qid': ['qid:0', 'qid:1','qid:0'],
'0': [0.4967, 0.4967,0.4967],
'1': [0.4967, 0.4967,0.4967],
'2': [0.4967, 0.4967,0.4967],
'3': [0.4967, 0.4967,0.4967],
'4': [0.4967, 0.4967,0.4967]}
df = pd.DataFrame(data)