如何将数据框中的行数写入 JSON 文件？

Posted 2023-04-15

【中文标题】如何将数据框中的行数写入 JSON 文件？

【英文标题】：How to write a count of rows in a dataframe to JSON files?

【发布时间】：2020-02-11 00:35:10

【问题描述】：

我有一个要求，我需要过滤数据帧，然后将返回的行数写入单独的 json 文件。 我正在通过以下方式过滤数据框：

 dfJson.where((col("Level") == "Medium")).select(dfJson["*")

我可以用 count 函数得到一个计数：

print(dfJson.count())

如何将此计数值保存到文本或 json 文件？

【参考方案1】：

你可以使用默认的python函数写入文件

f = open("test.txt", "a")
f.write(dfJson.count())
f.close()