I would like to compute this simple code in python, given a matrix modify it according to its entries. If the (i, j)-th
entry is greater than or equal to 1
then make it to the power of a
else make it 1
.
import numpy
def restricted_power(k, n, d, a):
"""
:param d: a distance matrix
:param k, n: shape of d
:param a: a positive real number
:return: a modified distance matrix
"""
x = numpy.zeros((k,n))
for i in range(k):
for j in range(n):
if d[i, j] < 1:
x[i, j] = 1
else:
x[i, j] = d[i, j] ** a
return x
Is there a way to code this without the loops ?
Aucun commentaire:
Enregistrer un commentaire