Question

i was helping my friend to do logic algorithm in python but i haven’t come with best solution yet.

first of all, i have a list of array:

x = array[0,1,2,3,4,3,2,3,-2,-4,-7,2,2]

and he wanted to categorize the x so the output become like this:

array([0,1,2,3,4]) # increasing value
array([4,3,2]) #decreasing value
array([2,3]) # increasing value
array([3,-2,-4,-7]) #decreasing value
array([-7,2]) # increasing value
array([2,2])  # remain_the_same_value

the rule is simple:

1. if the value keep increasing (like example above: 0,1,2,3,4) they put in one array
2. if the value keep decreasing (like example above: 3,-2,-4,-7) they put in one array
3. but, if there is a sudden change in value pattern such as example above: from the increasing value (0,1,2,3,4) suddenly the next value is decreasing. the new array will be made and put the last increasing value which is (4) and monitor the next value, whether it is decreasing value or not. If yes, they will be put in one array. example :array([4,3,2])
4. if the the value is remain the same (like example above, from 2 to 2). they will be put in one array.

this is what i come so far, but still far from the solution

#categorize which types of input
if len(x) > 2 :
    for i in range(len(x)) :
        if (x[i+1]-x[i]) > 0 and i+i < len(x) : # for increasing x value

        elif (x[i+1]-x[i]) < 0 and i+i < len(x) : # for decreasing x value

        elif (x[i+1]-x[i]) == 0 and i+i < len(x) : # for foward direction of vehicle

        else :
            print 'ERROR : check the input coordinates once again!'

best regards,

Glenn

Answer 1

Firstly I would like to say that I don’t understand one part of your question,

array([3,-2]) #decreasing value
array([-2,-4,-7]) #decreasing value

Why are these separate?

I will post my answer so far which gives the correct results except for that section since i don’t see the logic behind it. This example uses lists and tuples for simplicity but you can change it to use array if you want.

>>> from itertools import groupby
>>> data = [0,1,2,3,4,3,2,3,-2,-4,-7,2,2]
>>> def slope(a,b): #Returns 1 for inc, 0 for equal and -1 for dec
        return (a > b) - (a < b) 

>>> def groups(nums):
        for k,v in groupby(zip(nums,nums[1:]), lambda (x,y): slope(x,y)):
            yield next(v) + tuple(y for x,y in v) #Using itertools.chain this can be written as tuple(chain(next(v),(y for x,y in v)))


>>> list(groups(data))
[(0, 1, 2, 3, 4), (4, 3, 2), (2, 3), (3, -2, -4, -7), (-7, 2), (2, 2)]