Robot Framework DataDriver：如何从CSV随机选取单行数据

如果你的代码现在是遍历所有行重复执行测试，要改成随机选单行，这里有几种实用的实现方式，以Python为例：

方法一：直接读取所有行后随机选择（适合小到中型CSV，3000行完全适用） #

这种方式简单直观，把所有数据行读进内存后用random.choice()随机挑选一行，代码量少，执行效率也足够：

import csv
import random

def pick_random_row(csv_path):
    with open(csv_path, 'r', newline='', encoding='utf-8') as csv_file:
        reader = csv.reader(csv_file)
        # 跳过表头（不需要的话删掉这行）
        header = next(reader)
        rows = list(reader)
        
        if not rows:
            raise ValueError("CSV里没有数据行")
        
        random_row = random.choice(rows)
        # 把行数据和表头对应成字典，方便后续填入表单
        return dict(zip(header, random_row)) if header else random_row

# 调用示例
form_data = pick_random_row("your_table.csv")
# 这里直接用form_data填表单即可

方法二：流式随机选取（适合超大型CSV，节省内存） #

如果以后你的CSV行数远超3000，不想把所有行都读进内存，可以用蓄水池抽样算法，逐行读取时随机选中一行，最终得到的结果也是均匀随机的：

import csv
import random

def pick_random_row_large_file(csv_path):
    with open(csv_path, 'r', newline='', encoding='utf-8') as csv_file:
        reader = csv.reader(csv_file)
        header = next(reader)
        selected_row = None
        row_num = 0
        
        for row in reader:
            row_num += 1
            # 每一行有1/row_num的概率被选中，最终保证所有行概率均等
            if random.randint(1, row_num) == 1:
                selected_row = row
        
        if not selected_row:
            raise ValueError("CSV里没有数据行")
        
        return dict(zip(header, selected_row)) if header else selected_row

方法三：用Pandas快速实现（如果你已经在用数据分析库） #

如果你的项目里已经引入了Pandas，一行代码就能搞定随机选行，非常简洁：

import pandas as pd

df = pd.read_csv("your_table.csv")
# 随机选一行并转为字典
form_data = df.sample(n=1).iloc[0].to_dict()

不管用哪种方法，拿到form_data后直接把它的键值对填入表单对应的字段即可，这样就不会再遍历所有行重复执行测试了。

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