truncate appended return vectors

bnavigator · bnavigator · commit a6ea2968326a · 2021-03-26T14:44:51.000+01:00
diff --git a/control/timeresp.py b/control/timeresp.py
@@ -449,7 +449,7 @@ def forced_response(sys, T=None, U=0., X0=0., transpose=False,
                 if U.ndim == 1:
                     U = np.append(U, U[-1])
                 else:
-                    U = np.append(U, U[:, -1], 1)
+                    U = np.append(U, U[:, -1:], 1)
 
         else:
             sys_dt = dt         # For unspecified sampling time, use time incr
@@ -472,7 +472,8 @@ def forced_response(sys, T=None, U=0., X0=0., transpose=False,
         xout = np.transpose(xout)
         yout = np.transpose(yout)
 
-    return _process_time_response(sys, tout, yout, xout, transpose=transpose,
+    return _process_time_response(sys, tout[:n_steps], yout[:, :n_steps],
+                                  xout[:, :n_steps], transpose=transpose,
                                   return_x=return_x, squeeze=squeeze)
 
 
