修改程序for line in lines: values = line.split() i = int((int(float(values[0])) - col1_min) / interval_dx) j = int((int(float(values[1])) - col2_min) / interval_dy) if values[3] == '1': matrix[i][j] = 1 # 生成图片 img = Image.new('RGB', (width, height), black_color) for i in range(m): for j in range(n): if matrix[i][j] == 1: img.putpixel((i, j), red_color),在生成图片时,第四列数据为1时,将该像素点置为红色,第五列数据为1时,将该点的像素点为白色
时间: 2024-03-27 09:34:55 浏览: 11
修改后的程序如下:
```
# 初始化 matrix 矩阵
matrix = [[0] * n for i in range(m)]
# 读取并处理数据
for line in lines:
values = line.split()
i = int((int(float(values[0])) - col1_min) / interval_dx)
j = int((int(float(values[1])) - col2_min) / interval_dy)
if values[3] == '1':
matrix[i][j] = 1
# 生成图片
img = Image.new('RGB', (width, height), black_color)
for i in range(m):
for j in range(n):
if matrix[i][j] == 1:
if values[4] == '1':
img.putpixel((i, j), white_color)
else:
img.putpixel((i, j), red_color)
```
修改的部分是将生成图片的部分移动到了数据处理完毕的后面。在生成图片时,如果 matrix 矩阵中的元素为 1,就将该像素点设置为红色;如果第五列数据为 1,就将该像素点设置为白色。这样就可以根据第四列和第五列数据的不同,给不同的像素点设置不同的颜色。