How To Create An Array Using For Loop In Python