Numpy creation fromfunction

Question

I need something like this

def qqq(i,j):
    if i+j>2:
        return 0.5
    else:
        return 0
n=3
dcdt=np.fromfunction(lambda i,j: qqq(i,j)*i*j, (n,n), dtype=int)

but with more complicated qqq. But it leads to the error "The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()". I know the problem is that function is called once. How can I do such array creation with the "if-elif-else" structure in the function?

Jaime · Accepted Answer · 2014-02-28 08:41:43Z

2

You should turn your qqq function into something like:

def qqq(i, j):
    return np.where(i + j > 2, 0.5, 0)

See np.where's docs for details.

answered Feb 28, 2014 at 8:41

Jaime

67.7k19 gold badges128 silver badges164 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Numpy creation fromfunction

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related