如何将Python DataFrame中的array元组转为普通元组以实现批量插入？

我在Python的DataFrame中得到了一个包含numpy数组的元组，格式如下：

(array(['2023-04-01', '2023-06-01',
        'item1', 10, 2,
        'Promo1', 'NULL'], dtype=object),
 array(['2023-04-01', '2023-06-01',
        'item2', 10, 2,
        'Promo1', 'NULL'], dtype=object),
 array(['2023-04-01', '2023-06-01',
        'item3', 10, 4,
        'Promo1', 'NULL'], dtype=object),
 array(['2023-04-01', '2023-06-01',
        'item4', 10, 4,
        'Promo1', 'NULL'], dtype=object),
 array(['2023-04-01', '2023-06-01',
        'item5', 10, 2,
        'Promo1', 'NULL'], dtype=object))

希望将其转换为纯元组嵌套的格式：

(
   ('2023-04-01', '2023-06-01',
    'item1', 10, 2,
    'Promo1', 'NULL', ),
   ('2023-04-01', '2023-06-01',
    'item2', 10, 2,
    'Promo1', 'NULL', ),
   ('2023-04-01', '2023-06-01',
    'item3', 10, 4,
    'Promo1', 'NULL', ),
   ('2023-04-01', '2023-06-01',
    'item4', 10, 4,
    'Promo1', 'NULL', ),
   ('2023-04-01', '2023-06-01',
    'item5', 10, 2,
    'Promo1', 'NULL', )
)

本质上我要实现DataFrame数据的批量插入（比如每5行一批），已经能用iloc循环处理DataFrame，但卡在元组转换这一步，请问该怎么实现？

直接转换已有数组元组 #

如果已经拿到了包含numpy数组的元组，只需遍历每个数组并转为元组，再重新组合成大元组：

import numpy as np

# 原始包含数组的元组
original_data = (
    np.array(['2023-04-01', '2023-06-01', 'item1', 10, 2, 'Promo1', 'NULL'], dtype=object),
    np.array(['2023-04-01', '2023-06-01', 'item2', 10, 2, 'Promo1', 'NULL'], dtype=object),
    np.array(['2023-04-01', '2023-06-01', 'item3', 10, 4, 'Promo1', 'NULL'], dtype=object),
    np.array(['2023-04-01', '2023-06-01', 'item4', 10, 4, 'Promo1', 'NULL'], dtype=object),
    np.array(['2023-04-01', '2023-06-01', 'item5', 10, 2, 'Promo1', 'NULL'], dtype=object)
)

# 转换为嵌套元组格式
target_tuple = tuple(tuple(arr) for arr in original_data)

从DataFrame直接生成批量元组 #

如果是从DataFrame处理批量数据，无需生成中间数组元组，直接按批次提取行转换：

import pandas as pd

# 假设你的DataFrame为df
batch_size = 5
total_rows = len(df)

for start_idx in range(0, total_rows, batch_size):
    # 提取当前批次的行
    batch_df = df.iloc[start_idx:start_idx+batch_size]
    # 将批次数据转为嵌套元组
    batch_tuple = tuple(tuple(row) for _, row in batch_df.iterrows())
    # 此处可执行批量插入操作，比如传入数据库执行executemany
    # cursor.executemany(insert_sql, batch_tuple)

格式化输出（可选） #

如果需要像示例那样带换行的美观格式，使用pprint模块：

from pprint import pprint

pprint(target_tuple, width=40)

输出会自动按指定宽度换行，匹配你想要的展示格式。

内容的提问来源于stack exchange，提问作者Cignitor