pandas_stubs_dataframe_type_checking_with_mypy.py

1import pandas as pd
2
3# The pandas-stubs package provides type hints for pandas
4# This allows type checkers like mypy to catch errors in your code
5def process_data(df: pd.DataFrame) -> pd.Series:
6    # Type hints enable autocompletion and static analysis
7    result = df["value"] * 2
8    return result
9
10data = {"value": [1, 2, 3, 4, 5]}
11df = pd.DataFrame(data)
12
13# Static type checkers will now verify that 'df' is a DataFrame 
14# and the return type is a Series
15processed = process_data(df)
16print(processed)