返回信息流已有一个大小为200*300的一维数组数组，值为1,2,3,4，现在想将其可视化，输出为一个有四种颜色的200*300图片。已有思路为利用for 循环，将1,2,3,4分别以[R,G,B]的形式输入另一个200*300的三维数组，但是代码会一直报错，无法输入，求解答。代码如下，不知道怎么改了QAQ
B = np.zeros((223,387,3))
for i in pre:
for b in B:
if i == [4]:
b[i] ==[255,0,0]
break
if i == [3]:
b[i] ==[0,255,0]
break
if i == [2]:
b[i] == [0,0,255]
break
if i == [1]:
b[i] == [0,255,255]
break
print(B)
这是一条镜像帖。来源：北邮人论坛 / python / #18020同步于 2017/5/25
该镜像源已超过 30 天没有更新，可能在源站已被删除。
Python机器人发帖
【问题】如何用python给数组上色？
Prafor
2017/5/25镜像同步20 回复
订阅后，新回复会通过你的通知中心匿名送达。
9 条回复
py:32: VisibleDeprecationWarning: using a non-integer number instead of an integer will result in an error in the future
【 在 preytaren 的大作中提到: 】
: 贴一下报错信息吧
第一个是判断，第二个是赋值。改过来了
for i in pre:
for b in B:
if i == [4]:
b[i] =[255,0,0]
break
if i == [3]:
b[i] =[0,255,0]
break
if i == [2]:
b[i] =[0,0,255]
break
if i == [1]:
b[i] =[0,255,255]
break
print(B)
就是如果原来的值是1的话，B组同样位置就赋一个指定的的RGB的值。
【 在 preytaren 的大作中提到: 】
: 但是这i==[1]是什么鬼，其实直接在B上做矩阵运算比较好