Python comparison numpy matrix

Question

I need a numpy matrix with comparisons among all elements in a sequence:

from numpy import matrix
seq = [0, 1, 2, 3]
matrix([[cmp(a, b) for a in seq] for b in seq])

I'd like to know if there is a better way to do this.

Useful reference for anyone wanting to take this on: scipy.org/NumPy_for_Matlab_Users — Alain, Jul 6 '12 at 15:02

Winston Ewert · Accepted Answer · 2012-07-06 15:06:02Z

Why are you using matrix rather than array? I recommend sticking with array

As for a better way:

seq = numpy.array([0, 1, 2, 3])
numpy.clip(seq[:,None] - seq, -1, 1)

seq[:, None], moves the numbers into the second dimension of the array
seq[:, None] - seq broadcasts the numbers against each other, and subtracts them.
numpy.clip converts lower/higher values into 1 and -1

1 Answer