Question

I have an nxnxn matrix which I want to loop over and replace all values < 1E-35 with 1E-35.

          for i in range(N):
              for j in range(N):
                  for k in range(N):
                      if data[i][j][k] < 1E-35:
                          data[i][j][k] = 1E-35

Doesn’t seem to work.

Edit: I worked it out. It was both the indentation and incorrect indexing [i][j][k].

Problem remains: this loops from 0 to N-1 of the NxNxN data? When I do data.min() I still get values ~ 1E-101 which should be 1E-35 after the loop. Am I doing the for loop wrong?

Answer 1

It looks to me like you are using numpy in which case, you may want np.where:

data = np.where(data < 1e-35,1e-35,data)

Alternatively, you could use fancy indexing:

data[ data < 1e-35 ] = 1e-35