用值的均匀分布填充CSV

2024-04-23 13:44:10 发布

您现在位置:Python中文网/ 问答频道 /正文

我想写一个python脚本来填充基于列的CSV文件,并且每列都有一个只能在列中的预设字符串列表

示例

对于“name”列,唯一可能的值是“Andrew”、“James”、“Steve”、“Charlie”等,并且以类似的方式,其他每个列都有一个可能值的预设列表

我想在CSV中填充一个尽可能均匀的分布,这样就有了不同列值组合的行

问题:

  1. 我应该使用列表和随机选择值吗
  2. 会不会造成裁员
  3. 有没有更好的方法

Tags: 文件csv方法字符串name脚本示例列表
1条回答
网友
1楼 · 发布于 2024-04-23 13:44:10

以下内容将帮助您开始:

import csv
import random

with open('output.csv', 'w', newline='') as f_output:
    csv_output = csv.writer(f_output)
    csv_output.writerow(["Name", "Age"])

    names = ["Andrew", "Steve", "Charlie"]

    for _ in range(10):     # Create 10 entries
        name = random.choice(names)
        age = random.randint(20, 50)

        csv_output.writerow([name, age])

提供示例输出,例如:

Name,Age
Andrew,50
Steve,37
Charlie,34
Andrew,47
Andrew,28
Andrew,27
Charlie,29
Charlie,49
Steve,42
Charlie,38

相关问题 更多 >