我需要一个三维嵌套阵列。如何将其展平为1d阵列?(因为性能)我假设,我需要乘以一些东西等等,但我不是最伟大的数学家。 谢谢
def imgRGB(im):
width, height = im.size
#create matrix
matrix = []
for i in range(height):
matrix.append([[0,0,0] for _ in range(width)])
#get the rgb values
for x in range(width):
for y in range(height):
r,g,b = im.getpixel((x,y))
matrix[y][x][0] = r
matrix[y][x][1] = g
matrix[y][x][2] = b
return matrix
像素值将与其他RGB值进行比较,以找到最接近的匹配
def getNearestColor(rgb):
a = []
for i in range(len(rgbValues)):
d = ((rgbValues[i][0]-rgb[0])*0.3)**2 + ((rgbValues[i][1]-rgb[1])*0.59)**2 + ((rgbValues[i][2]-rgb[2])*0.11)**2
a.append(d)
list.sort(a)
return a[0]
如果我理解正确,您需要执行以下操作:
其中
rgb
是一些值r、g、b的列表[r, g, b]
您可以使用ndarray.flatte,在您的示例中是
matrix.flatten()
。这将返回矩阵的展平版本。文档here也有一个例子相关问题 更多 >
编程相关推荐