How to make if else condition in python 2d array

Question

I have a 2d array with shape(3,6), then i want to create a condition to check a value of each array. my data arry is as follows :

array([[ 1, 2, 3, 4, 5, 6], 7, 8, 9, 10, 11, 12], [13, 14, 15, 16, 17, 18]])

if in an array there are numbers < 10 then the value will be 0

the result I expected

array([[ 0, 0, 0, 0, 0, 0], 0, 0, 0, 10, 11, 12], [13, 14, 15, 16, 17, 18]])

the code i created is like this, but why can't it work as i expected

FCDataNew = []

a = [ [1,2,3,4,5,6], 
     [7,8,9,10,11,12], 
     [13,14,15,16,17,18]
     ]

a = np.array(a)

c = 0
c = np.array(c)

for i in range(len(a)):
  if a[i].all()<10:
    FCDataNew.append(c)
  else:
    FCDataNew.append(a[i])

FCDataNew = np.array(FCDataNew)
FCDataNew

mozway · Accepted Answer · 2022-11-21 12:35:34Z

2

If you want to modify the array in place, use boolean indexing:

FCDataNew = np.array([[1,2,3,4,5,6],
                      [7,8,9,10,11,12],
                      [13,14,15,16,17,18],
                     ])

FCDataNew[FCDataNew<10] = 0

For a copy:

out = np.where(FCDataNew<10, 0, FCDataNew)

Output:

array([[ 0,  0,  0,  0,  0,  0],
       [ 0,  0,  0, 10, 11, 12],
       [13, 14, 15, 16, 17, 18]])

edited Nov 21, 2022 at 12:35

answered Nov 21, 2022 at 12:28

mozway

267k13 gold badges56 silver badges106 bronze badges

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

11 Comments

stack offer Over a year ago

FCDataNew = [] a = [ [1,2,3,4,5,6], [7,8,9,10,11,12], [13,14,15,16,17,18] ] a = np.array(a) c = 0 c = np.array(c) for i in range(len(a)): if a[i].all()<10: FCDataNew.append(c) else: FCDataNew.append(a[i]) FCDataNew = np.array(FCDataNew) out = np.where(FCDataNew<10, 0, FCDataNew)

stack offer Over a year ago

Is it like this?

mozway Over a year ago

No, just what I wrote in the answer, no need for a loop

stack offer Over a year ago

but how if in the case I have negative value?

stack offer Over a year ago

array([[ -8, -2, 3, 4, -1, 2], [ 7, -1, -2, -1, 10, -3], [13, 14, 15, 16, 17, 18]])

|

uozcan12 · Accepted Answer · 2022-11-21 12:31:26Z

0

You can just use arr[arr < 10] = 0

answered Nov 21, 2022 at 12:31

uozcan12

4251 gold badge5 silver badges13 bronze badges

Collectives™ on Stack Overflow

How to make if else condition in python 2d array

2 Answers 2

11 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

11 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related