Python: Sort an unsorted array numbers using Wiggle sort

Python Search and Sorting: Exercise-24 with Solution

Write a Python program to sort an unsorted array of numbers using Wiggle sort.

Wiggle Sort:
Given an unsorted array nums, reorder it in-place such that nums[0] <= nums[1] >= nums[2] <= nums[3]....
For example, given nums = [3, 5, 2, 1, 6, 4], one possible answer is [1, 6, 2, 5, 3, 4].

Sample Solution:

Python Code:

def wiggle_sort(arra_nums):
    for i, _ in enumerate(arra_nums):
        if (i % 2 == 1) == (arra_nums[i - 1] > arra_nums[i]):
            arra_nums[i - 1], arra_nums[i] = arra_nums[i], arra_nums[i - 1]

    return arra_nums

print("Input the array elements: ")
arra_nums = list(map(int, input().split()))
print("Original unsorted array:")
print(arra_nums)
print("The said array after applying Wiggle sort:")
print(wiggle_sort(arra_nums))

Sample Output:

Input the array elements: 
 1 5 2 3 4 6
Original unsorted array:
[1, 5, 2, 3, 4, 6]
The said array after applying Wiggle sort:
[1, 5, 2, 4, 3, 6]

Flowchart:

Python Code Editor:

Contribute your code and comments through Disqus.

Previous: Write a Python program to sort a list of elements using Tree sort.
Next: Write a Python program to sort unsorted numbers using Timsort.