python - 从 python 数组中选择行

Question

我有两个数组。假设它们看起来像这样:

    time1 = [ 1 2 3 ] and time2 = [ 2 4 6]
            [ 4 5 6 ]         
            [ 7 8 9 ]         

我只想选择 time1 中第一列在 time2 范围内的行。例如，在此数据集中，我将绘制 [4 5 6] 行，因为 4 在 2 - 6 的范围内。我尝试从数组 time1 中选择行，如下所示:

selectedtimes = time1(any(time1[:,0] < time2[-1]) and any(time1[:,0] > time2[0]))

我当前收到对象不可调用错误(如下所示)，并且陷入困境。有没有更好的方法来重写这一行？

'numpy.ndarray' object is not callable

感谢帮助!

Answer 1

您可以使用numpy.logical_and这里:

>>> np.logical_and(time1[:,0] > time2[0], time1[:,0] < time2[-1] )
array([False,  True, False], dtype=bool)
>>> time1[np.logical_and(time1[:,0] > time2[0], time1[:,0] < time2[-1] )]
array([[4, 5, 6]])