/ BigData

Generating Sensible Transaction Data with Python

The other day, I was facing a scenario where I had to setup bucketing with Hive and I needed some sample data, but in the same way I thought it would've been nice to have some random data, but that makes sense, and not just have random data that does not make sense.

I ended up writing a Python Script that generates comma delimited lines that represents dummy transactional data that ends up like the following:

#AccId, Date, TransactionId, Price, MerchantName, MerchantCategory, City, Province, Purchase Method
0714,2015-07-01,4000645,506.94,Rage Shoes,Shoes,Kroonstad,Free State,credit
``` <p>

Here is the Python Script that will generate 1000 lines per run:

```language-python
import random

method = ["credit", "debit", "credit", "credit", "debit", "credit", "credit", "cash", "credit"]

shop_dict = {
    'Edgars': 'Clothing',
    'CNA': 'Stationary',
    'Sportmans Warehouse': 'Sports Equipment',
    'Pick n Pay': 'Groceries',
    'Rage Shoes': 'Shoes',
    'BT Games': 'Games',
    'Dions Wired': 'Electronics',
    'Makro': (
        'Shoes',
        'Clothing',
        'Electronics',
        'Groceries',
        'Stationary',
        'Sports Equipment',
        'Games'
        )
    }

city_dict = {
    'Kroonstad': 'Free State',
    'Cape Town': 'Western Cape',
    'Pretoria': 'Gauteng',
    'PE': 'Eastern Cape',
    'Johannesburg': 'Gauteng',
    'Potchefstroom': 'North West',
    'Bloemfontein': 'Free State',
    'Stellenbosch': 'Western Cape',
    'Krugersdorp': 'Gauteng',
    'Somerset West': 'Western Cape',
    'Kimberley': 'Northern Cape',
    'Upington': 'Northern Cape',
    'Klerksdorp': 'North West',
    'Potchefstroom': 'North West',
    'Parys': 'Free State',
    'Gauteng': 'Boksburg'
    }


for x in xrange(10000):
    x_stores = random.choice(shop_dict.keys())
    if x_stores == 'Makro':
        x_category = shop_dict[x_stores][random.randint(0,6)]
    else:
        x_category = shop_dict[x_stores]

    x_cities = random.choice(city_dict.keys())
    x_province = city_dict[x_cities]

    print (
        str(random.randint(1, 1000)).zfill(4) + ',' +
        str(random.randint(2014,2016)) + '-' + str(random.randint(1,12)).zfill(2) + '-' + str(random.randint(1,28)).zfill(2) + ',' +
        str(random.randint(4000000,4001000)) + ',' +
        str(random.randint(29, 800)) + '.' + str(random.randint(0, 99)).zfill(2) + ',' +
        x_stores + ',' +
        x_category + ',' +
        x_cities + ',' +
        x_province + ',' +
        random.choice(method)
    )

# Schema: AccountID, DateOfPurchase, TransactionID, PurchasePrice, StoreName, StoreCategory, CityName, ProvinceName, PurchaseMethod
``` <p>

After generating a couple of outputs, I loaded them to HDFS, created a table and mapped the data and then I was able to what was expected.