如何在 PySpark 中将字典转换为数据框

Question

我有输入

{'A':'1','B':'1'}

我需要的输出

+----+----+
|A   |B   |
+----+----+
|   1|   2|
+----+----+

我试过的代码如下。这不起作用

v = {'A': '1','B':'2'}

rdd = sc.parallelize(v.values())

df = spark.createDataFrame(rdd, list(v.keys()))
df.show()

Answer 1

我找到了一种使用 json 解析器的简单方法。

spark = SparkSession.builder.appName('abc').enableHiveSupport().getOrCreate()
sc = spark.sparkContext
dict_lst = {'A': '1','B':'2'}

rdd = sc.parallelize([dict_lst])
print(type(rdd))
df = spark.read.json(rdd)

df.show()