Заполнение CSV с равномерным распределением значений - PullRequest
0 голосов
/ 04 июля 2019

Я хочу написать скрипт Python для заполнения файла CSV на основе столбцов, и каждый столбец имеет предустановленный список строк, которые могут быть только в столбце.

Пример :

Для столбца «имя» единственными возможными значениями являются «Эндрю», «Джеймс», «Стив», «Чарли» и т. Д., И аналогичным образом каждый из остальных столбцов имеет предустановленный список возможных значений..

Я бы хотел заполнить CSV в максимально равномерном распределении, чтобы были строки различных комбинаций значений столбцов.

Вопрос:

  1. Стоит ли использовать списки и случайным образом выбирать значения?
  2. Это приведет к избыточности?
  3. Есть ли более эффективные способы сделать это?

1 Ответ

0 голосов
/ 04 июля 2019

Следующее должно помочь вам начать:

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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...