Conditional operations on numpy arrays

后端 未结 3 1706
深忆病人
深忆病人 2021-01-04 21:48

I\'m new to NumPy, and I\'ve encountered a problem with running some conditional statements on numpy arrays. Let\'s say I have 3 numpy arrays that look like this:

a:

3条回答
  •  隐瞒了意图╮
    2021-01-04 22:21

    A slight change in the numpy expression would get the desired results:

    c += ((a > 3) & (b > 8)) * b*2
    

    Here First I create a mask matrix with boolean values, from ((a > 3) & (b > 8)), then multiply the matrix with b*2 which in turn generates a 3x4 matrix which can be easily added to c

提交回复
热议问题