ronniebasak's picture
Push data
7d5997c
import csv
import random
def generate_random_names(count):
first_names = ["Alex", "Jordan", "Taylor", "Morgan", "Casey", "Jamie", "Riley", "Cameron", "Drew", "Logan", "Priya", "Mia", "Monica"]
last_names = ["Smith", "Johnson", "Brown", "Taylor", "Anderson", "Lee", "Garcia", "Martinez", "Davis", "Miller", "Chaddha", "Singhania", "Das"]
return [f"{random.choice(first_names)} {random.choice(last_names)}" for _ in range(count)]
def replace_users_with_names(input_file, output_file):
with open(input_file, 'r') as csvfile:
reader = csv.reader(csvfile)
rows = list(reader)
# Generate random names for all rows needing replacements
random_names = generate_random_names(len(rows))
# Replace "User is" with random names
updated_rows = []
for i, row in enumerate(rows):
if "User is" in row[1]:
row[1] = row[1].replace("User is", random_names[i] + " is")
updated_rows.append(row)
# Write the updated rows to a new CSV
with open(output_file, 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(updated_rows)
# Usage example
replace_users_with_names('prompts_final.csv', 'prompts_final2.csv')