如何在Airflow中的任务之间传递数据帧

from airflow import DAG
from airflow.operators.python_operator import PythonOperator
import pandas as pd

def generate_dataframe():
    # 生成数据帧
    df = pd.DataFrame({'col1': [1, 2, 3], 'col2': ['a', 'b', 'c']})
    # 将数据帧存储到XCom中
    return df.to_dict()

with DAG('dataframe_dag', schedule_interval=None, start_date=datetime(2022, 1, 1)) as dag:
    task_generate_dataframe = PythonOperator(
        task_id='generate_dataframe',
        python_callable=generate_dataframe,
        provide_context=True
    )

from airflow import DAG
from airflow.operators.python_operator import PythonOperator
import pandas as pd

def process_dataframe(**context):
    # 从XCom中获取数据帧
    df_dict = context['ti'].xcom_pull(task_ids='generate_dataframe')
    df = pd.DataFrame.from_dict(df_dict)
    # 对数据帧进行处理
    # ...

with DAG('dataframe_dag', schedule_interval=None, start_date=datetime(2022, 1, 1)) as dag:
    task_generate_dataframe = PythonOperator(
        task_id='generate_dataframe',
        python_callable=generate_dataframe,
        provide_context=True
    )

    task_process_dataframe = PythonOperator(
        task_id='process_dataframe',
        python_callable=process_dataframe,
        provide_context=True
    )

    task_generate_dataframe >> task_process_dataframe