首页python numpy 取满足条件的行

python numpy 取满足条件的行

时间: 2023-07-25 08:12:26 浏览: 145

从numpy数组中取出满足条件的元素示例

例如问题：从 arr 数组中提取所有奇数元素。 input：arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) output: #> array([1, 3, 5, 7, 9]) Solution: #Input >>> arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) #Solution arr = arr[arr % 2 == 1] >>> array([1, 3, 5, 7, 9]) 这是对奇数元素的提取，其他条件同理，在这个问题上我们深入再学习一下，首先 numpy 中的 array 数组是可以直

可以使用numpy的布尔索引来取满足条件的行。假设有一个numpy数组`arr`和一个条件`cond`，可以使用以下代码来取满足条件的行： ```python selected_rows = arr[cond] ``` 其中，`cond`是一个布尔数组，与`arr`的行数相同，表示每一行是否满足条件。`selected_rows`是一个新的numpy数组，包含满足条件的所有行。例如，如果要取`arr`中第二列大于0的所有行，可以使用以下代码： ```python selected_rows = arr[arr[:, 1] > 0] ``` 这里使用了布尔索引`arr[:, 1] > 0`，表示取`arr`的第二列大于0的所有行。

阅读全文