如何在pyspark中创建包含两个dataframe列的字典？

from pyspark.sql import SparkSession
from pyspark.sql.functions import col

spark = SparkSession.builder.getOrCreate()

df1 = spark.createDataFrame([(1, 'John'), (2, 'Alice'), (3, 'Bob')], ['id', 'name'])
df2 = spark.createDataFrame([(1, 'USA'), (2, 'Canada'), (3, 'UK')], ['id', 'country'])

df_dict = df1.select('id', 'name').rdd.collectAsMap()
df_dict.update(df2.select('id', 'country').rdd.collectAsMap())