如何对numpy arrary中的数据进行墙砖
时间: 2023-11-23 11:07:25 浏览: 29
首先,可以使用numpy中的reshape()函数将数组变形为二维数组,然后利用for循环遍历每个元素,并根据它的值决定墙砖的颜色和位置。可以使用matplotlib库中的imshow()函数将二维数组渲染成墙砖效果。
下面是一个示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成一个随机数组
arr = np.random.randint(0, 3, size=(10, 10))
# 将数组变形为二维数组
arr = arr.reshape(-1)
# 创建一个画布
fig = plt.figure(figsize=(10, 10))
# 遍历每个元素,并根据其值绘制不同颜色的砖块
for i in range(len(arr)):
if arr[i] == 0:
color = 'gray'
elif arr[i] == 1:
color = 'red'
else:
color = 'blue'
x = i % 10
y = i // 10
plt.fill([x, x+1, x+1, x], [y, y, y+1, y+1], color=color)
# 显示图片
plt.axis('off')
plt.show()
```
这段代码会生成一个大小为10x10的随机数组,并将其渲染成墙砖效果,其中0表示灰色砖块,1表示红色砖块,2表示蓝色砖块。可以根据需要修改颜色和数组大小等参数。