使用pyarrow重新划分pyarrow表的大小，并将其写入几个拼图文件？

import pyarrow as pa
import pyarrow.parquet as pq

table = pa.Table.from_pandas(df)

chunk_size = 1000000  # 设置每个拼图文件的大小
num_chunks = len(table) // chunk_size + 1  # 计算拼图文件的数量

# 将表按照指定大小划分为多个拼图文件
chunks = [table[i*chunk_size:(i+1)*chunk_size] for i in range(num_chunks)]

output_dir = 'path/to/output/directory'  # 设置输出目录

# 将每个拼图文件写入磁盘
for i, chunk in enumerate(chunks):
    output_file = f'{output_dir}/chunk_{i}.parquet'
    pq.write_table(chunk, output_file)