Python3 Compatibily for data generation

[hmrishavbandy]

In Python 3, dict.keys doesn't return a list, but something like a set that doesn't support indexing. To make indexing possible, it is necessary to convert that to a list.

[CLAassistant]

All committers have signed the CLA.