pyarrow-stubs

import pyarrow as pa from typing import List def create_arrow_table(names: List[str], ages: List[int]) -> pa.Table: """Creates a PyArrow Table from names and ages.""" if not (len(names) == len(ages)): raise ValueError("Lengths of names and ages must match.") # Create PyArrow Arrays name_array = pa.array(names, type=pa.string()) age_array = pa.array(ages, type=pa.int64()) # Create a PyArrow Table table = pa.table({'name': name_array, 'age': age_array}) return table if __name__ == "__main__": my_names = ["Alice", "Bob", "Charlie"] my_ages = [30, 24, 35] # This call would be type-checked by tools like MyPy/Pyright person_table: pa.Table = create_arrow_table(my_names, my_ages) print("Created PyArrow Table:") print(person_table) print(f"\nFirst person's name: {person_table.column('name')[0].as_py()}")