define_function_af.py

from collections import Counter

# define the python function
def my_function():
    # get the variable value
    file_path = Variable.get("data_path")
    # open the file
    file_ = open(file_path)
    # read the file and calculate the word count
    data = Counter(file_.read().split())
    return data

define_the_task_af.py

# define the DAG
dag = DAG(
    'word_count_example',
    default_args=default_args,
    description='How to use the Python Operator?',
    schedule_interval=timedelta(days=1),
)

# define the first task
t1 = PythonOperator(
    task_id='print_word_count',
    python_callable= my_function,
    dag=dag,
)

t1

variables_airflow.py

### importing the required libraries
from datetime import timedelta
from airflow import DAG
from airflow.operators.python_operator import PythonOperator
from airflow.utils.dates import days_ago
from airflow.models import Variable