Getting Array Input In Python Programming