Getting Array Input In Python Integer Functions