pandas loc fix

ROBelgium · Oct 18, 2021 · 323c53d · 323c53d
1 parent d38daf9
commit 323c53d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/msnoise/s06compute_dtt.py b/msnoise/s06compute_dtt.py
@@ -228,9 +228,8 @@ def main(interval=1, loglevel="INFO"):
                                 tindex = np.where((tArray >= rmlag) & (tArray <= rMlag))[0]
 
                             tmp = np.setdiff1d(np.arange(len(tArray)),tindex)
-                            df['err'][tmp] = 1.0
-                            df['coh'][tmp] = 0.0
-
+                            df.loc[:, 'err'].iloc[tmp] = 1.0
+                            df.loc[:, 'coh'].iloc[tmp] = 0.0
                             if first:
                                 tArray = df.index.values
                                 dtArray = df['dt']