python 用参数传递代替for循环的方法 将1维数组转换成2维RGB图像
时间: 2024-05-29 15:12:15 浏览: 155
python进行参数传递的方法
可以使用numpy的reshape函数将一维数组转换成二维数组,然后使用PIL库中的Image.fromarray函数将二维数组转换成RGB图像。
示例代码如下:
```python
import numpy as np
from PIL import Image
# 一维数组
arr1d = np.array([255, 0, 0, 0, 255, 0, 0, 0, 255])
# 转换成二维数组
arr2d = arr1d.reshape((3, 3))
# 转换成RGB图像
img = Image.fromarray(arr2d.astype('uint8'), mode='RGB')
img.show()
```
运行结果为:
![image](https://user-images.githubusercontent.com/26833433/110943310-1d8a5200-8372-11eb-9ec1-1a9d9a31fc24.png)
阅读全文