how to do the mapping btw numpy arrayvalues and matrix columns

devnew at gmail.com devnew at gmail.com
Sun Nov 4 12:22:18 EST 2007


CW,
thanx for the reply..but i was looking for a mapping BTW each item of
a numpy.ndarray and the corresponding column of a numpy.matrix ,after
some struggle :-) i came up with this

#a function to return a column from a matrix
def getcol(data, colindex):
    return data[:,colindex]    #returns a matrix obj

#function to set a column of a matrix with a given matrix
def setcol(data, inmat,colindex):
    data[:,colindex]=inmat  #both data and inmat are matrix objs

#now i have an ndarray with 5 elements
evalarray=array(([11.0,33.0,22.0,55.0,44.0]))

#and a matrix with 5 columns
evectmat=matrix(([1.3,2.5,3.2,6.7,3.1],
                  [9.7,5.6,4.8,2.5,2.2],
                  [5.1,3.7,9.6,3.1,6.7],
                  [5.6,3.3,1.5,2.4,8.5]

           ))

the first column of evectmat corresponds to first element of evalarray
and so on..
then i did this

mydict=dict(
[(evalarray[x],getcol(evectmat,x)) for x in range(len(evalarray))]
)
klst=mydict.keys()

klst.sort()
klst.reverse() #because i want the largest value as first

newevectmat=matrix(zeros((4,5)))

for x in range(len(klst)):
    newcol=mydict[klst[x]]
    setcol(newevectmat,newcol,x)

print "newevectmat:"
print newevectmat

this gives me the desired result..now i have a new matrix with columns
arranged corresponding to the values of evalarray

i don't know if this is a good way(or even pythonish ) to do it..i am
a beginner afterall.!. any suggestions most welcome
TIA
dn










>
> from numpy import matrix, asarray
> obj = matrix(([1.3,2.5,3.2,6.7,3.1],
>                [9.7,5.6,4.8,2.5,2.2],
>                [5.1,3.7,9.6,3.1,6.7],
>                [5.6,3.3,1.5,2.4,8.5]))
> ar = asarray(obj)
> val_to_col_list = []
> for row in ar:
>      for ind,val in enumerate(row):
>          val_to_col_list.append((val,ind))
> val_to_col_list.sort()
>
> If instead you require a map, such that each value maps to a list of the
> columns it appears in, you could try the following:
>
> val_col_map = {}
> for row in ar:
>      for col,val in enumerate(row):
>          tmplist=val_col_map.get(val,[])
>          tmplist.append(col)
>          val_col_map[val]=tmplist
> val_keys = val_col_map.keys()
> val_keys.sort()
>
> val_keys is now a sorted list of unique values from your original
> matrix.  Use these values as keys for val_col_map.
>
> Eww... but it works.  You'll want to be really careful with floating
> point numbers as keys, seehttp://docs.python.org/tut/node16.htmlfor
> more details.
>
> Best of luck,
>
> Cameron.





More information about the Python-list mailing list