pandas 将json文件转换为labels.csv

hivapdat  于 2023-04-19  发布在  其他
关注(0)|答案(1)|浏览(101)

我有labels.json文件包含图像名称和地面真值。由于库中的更改,我必须修改json文件中的数据

Json内部

{"자랑스럽다_2730052.jpg": "자랑스럽다", "만족스럽다_1299150.jpg": "만족스럽다"}

我想生成一个labels.csv文件,其中包含filename列和words列以及如下所示的labels.csv

filename        words
2730052.jpg     자랑스럽다

我该怎么做

w51jfk4q

w51jfk4q1#

你可以创建一个像{colname:[...]}这样的字典,然后把它变成dataframe,用r'.+_'替换文本,然后保存到csv文件。

json = {"자랑스럽다_2730052.jpg": "자랑스럽다", "만족스럽다_1299150.jpg": "만족스럽다"}
df = pd.DataFrame({'filname': json.keys(), 'words': json.values()})
df['filname'] = df['filname'].replace(r'.+_', '', regex=True)
df.to_csv('labels.csv', index=False)

df:

filname  words
0  2730052.jpg  자랑스럽다
1  1299150.jpg  만족스럽다

相关问题