numpy编程小技巧——向量运算

1. numpy库，可以进行大量的向量运算编程，而且编程代码非常简洁。

示例代码：

import numpy as np
import colorsys

color_mask = np.array([[217,200,205,209],[200,204,209,201],[220,221,222,227]])
print("color_mask:", color_mask)
id_train = {0: [0, 249, 255, 213, 206, 207, 211, 208, 216, 215, 218, 219, 232, 202, 231, 230, 228, 229, 233, 212, 223],
            1: [200, 204, 209], 2: [201, 203], 3: [217], 4: [210], 5: [214],
            6: [220, 221, 222, 224, 225, 226], 7: [205, 227, 250]}
encoder_mask = np.zeros((color_mask.shape[0],color_mask.shape[1]))
print("encoder_mask", encoder_mask)
for i in range(8):
    for item in id_train[i]:
        #color_mask和encoder_mask的维度相同，然后可以根据color_mask的值直接替换encoder_mask的值
        encoder_mask[color_mask == item] = i


print("new_encoder_mask:",encoder_mask)