Как записать количество строк в кадре данных в JSON файлов?

Question

У меня есть требование, где мне нужно отфильтровать фрейм данных, а затем записать количество строк, возвращающихся в отдельный файл json. Я фильтрую фрейм данных следующим образом:

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

Я могу получить счет с функцией подсчета как:

print(dfJson.count())

Как сохранить это значение счетчика в текст или json файл?

Answer 1

Вы можете использовать функцию по умолчанию python для записи в файл

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