from random import choice from csv import DictWriter COLS_NAMES = ['prticipant.label', 'attitude'] PARTICIPANTS_LABELS = list(range(1, 100)) ATTITUDES = ['pro', 'anti', 'neutral'] data = list() for label in PARTICIPANTS_LABELS: data.append( {COLS_NAMES[0]: label, COLS_NAMES[1]: choice(ATTITUDES)} ) with open('data_attitude.csv', 'w', newline='') as f: writer = DictWriter(f, fieldnames=COLS_NAMES) writer.writeheader() writer.writerows(data)