How To Create An Array Of Size N In Python