I am somehow new to python and would like to get rid of use of for loop as it makes the program very sl