import numpy as np rgb_list = np.array([[20, 81, 124], [67, 205, 128], [216, 56, 58], [243, 210, 102], [44, 62, 80]]) img_imerged_3d = np.array([[[54, 81, 124], [67, 205, 128], [216, 56, 58], [243, 210, 102]], [[44, 62, 80],[67, 205, 128], [216, 56, 58], [243, 210, 102]], [[67, 205, 128], [216, 56, 58], [243, 210, 102], [67, 205, 128]],[[216, 56, 58], [243, 210, 102],[67, 205, 128], [216, 56, 58]]]) #print(rgb_list.dtype) #print(rgb_list) #print(img_imerged_3d.dtype) #print(img_imerged_3d) img_imerged_2d = img_imerged_3d.reshape(-1,3) #print(img_imerged_2d) img_imerged_2d_rows = np.apply_along_axis(lambda x: x.tobytes(), 1, img_imerged_2d) rgb_list_rows = np.apply_along_axis(lambda x: x.tobytes(), 1, rgb_list) diff_rows = np.setdiff1d(img_imerged_2d_rows, rgb_list_rows) result = np.array([np.frombuffer(row, dtype=img_imerged_2d.dtype) for row in diff_rows]) print(result)
时间: 2023-02-11 10:17:40 浏览: 136
python 3.74 运行import numpy as np 报错lib\site-packages\numpy\__init__.py
这是一段 Python 代码,它定义了两个 numpy 数组:rgb_list 和 img_imerged_3d。
rgb_list 数组是一个二维数组,其中包含了 5 个由三个元素(R,G,B值)组成的数组
img_imerged_3d 数组是一个三维数组,其中包含了4个二维数组,每个二维数组包含4个三元素数组。
阅读全文