请用python帮我写一段眼罩建模代码
时间: 2024-12-13 22:24:34 浏览: 12
当然可以,这里是一个简单的Python模型,用于模拟随机生成眼罩图案。这个例子会创建一个二维数组表示眼罩,并填充黑色(0)或白色(1),以模拟遮住的部分:
```python
import numpy as np
def create_mask(size=50, pattern='random'): # 默认大小50x50,随机模式
"""
创建眼罩模型的函数
:param size: 图像尺寸,默认为50
:param pattern: 可选模式,支持 'random' (随机) 或 'stripes' (条纹)
:return: 眼罩图案矩阵
"""
if pattern == 'random':
mask = np.random.randint(0, 2, size=(size, size), dtype=np.bool)
elif pattern == 'stripes':
mask = np.zeros((size, size))
mask[::2, ::2] = 1 # 水平和垂直方向上交替放置1
else:
raise ValueError("Pattern should be either 'random' or 'stripes'.")
return mask
# 示例用法
mask = create_mask(pattern='random')
print("随机眼罩:")
print(mask)
mask = create_mask(pattern='stripes')
print("\n条纹眼罩:")
print(mask)
阅读全文