python生成一个6行6列的数组,随机数为1到100,删除包含小于5的元素行
时间: 2024-03-11 10:48:20 浏览: 83
向一个数组中插入一个1~100的随机数
可以使用numpy库生成随机数数组,并利用numpy的条件筛选功能来删除包含小于5的元素行。以下是代码示例:
```python
import numpy as np
arr = np.random.randint(1, 101, (6, 6))
print("原始数组:")
print(arr)
arr = arr[~np.any(arr < 5, axis=1)]
print("删除包含小于5的元素行后的数组:")
print(arr)
```
解释一下代码:
- `np.random.randint(1, 101, (6, 6))` 表示生成一个6行6列的数组,随机数范围为1到100。
- `arr[~np.any(arr < 5, axis=1)]` 利用numpy的条件筛选功能,删除包含小于5的元素行。其中 `~np.any(arr < 5, axis=1)` 的意思是找出行中所有元素都大于等于5的行,用 `~` 取反后即为找出行中包含小于5的元素的行。
- `print(arr)` 则是将删除元素后的数组打印输出。
希望这个回答能够帮到你!
阅读全文