partition_pseudo_code.py

# Chunk size for 104 sections
chunk_size = len(clients) // 103

# Create list of lists of clients in each section
client_lists = [clients[i:i + chunk_size] for i in range(0, len(clients), chunk_size)]

def create_partition(client_list, partition_number):
    """Creates and saves a dataset with users in user_list"""
    
    # Subset data 
    subset = data[client_list]
    
    # Save subset data with partition number as identifier
    subset.to_csv('p%s.csv' % partition_number)

# Iterate through each sublist of clients
for i, client_list in enumerate(client_lists):
    create_partition(client_list, i)