python使用Numpy创建白色图片

import numpy as np
import cv2

    img = np.zeros( [ 480, 640 ], dtype = np.uint8 )    # 数组元素是0，生成黑色的图片

    dst = np.zeros( ( 480, 640, 3 ), np.uint8 )

    # 转换二维数组的坐标值，改变图片颜色
    for i in range( 0, 480 ):
        for j in range( 0, 640 ):
            gray2 = 255                              # 数组元素是255，生成白色的图片
            dst[ i, j ] = np.uint8( gray2 )

    img = dst

    # 保存图片
    imgName = 'a1.png'
    cv2.imwrite( imgName, img, [ int( cv2.IMWRITE_PNG_COMPRESSION ), 4 ] )