在2D numpy数组中随机选择索引，了解ChatGPT解决方案

bvpmtnay 于 2024-01-08 发布在其他

关注(0)|答案(1)|浏览(186)

我让ChatGPT帮助我在加载为numpy数组的图像中随机选择像素。它给我的解决方案实际上工作得很好，但我很难理解它是如何工作的：

from PIL import Image
import numpy as np
img = Image.open('image.png')
pix = np.array(img)
random_pixels = np.random.choice(pix.shape[0]*pix.shape[1], pix.shape[0]*pix.shape[1], replace=False)
selected_pixels = []
for p in random_pixels:
    x = p // pix.shape[1]
    y = p % pix.shape[1]
    selected_pixels.append((x, y))

字符串
我不明白如何整数除法和模运算符应用于图像宽度实际上能够返回所有的坐标。

numpy

来源：https://stackoverflow.com/questions/77631136/randomly-selecting-indices-in-a-2d-numpy-array-understanding-chatgpt-solution

1条答案

按热度按时间

30byixjq1#

从一个可以实际检查的小数组开始

In [403]: pix=np.arange(12).reshape(3,4)
In [404]: pix
Out[404]: 
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
In [405]: random_pixels = np.random.choice(pix.shape[0]*pix.shape[1], pix.shape[0]*pix.shape[1], replace=False)
     ...: print(random_pixels)
     ...: selected_pixels = []
     ...: for p in random_pixels:
     ...:     x = p // pix.shape[1]
     ...:     y = p % pix.shape[1]
     ...:     print(p,x,y)
     ...:     selected_pixels.append((x, y))
     ...:

字符串
数组是（3，4）形状，choices随机创建了0到11的值。

[ 5  6  9  4  2  8 11 10  0  7  1  3]

型
迭代获取值p，并从中生成x,y坐标。

型
实际上，我不需要打印x，y，因为它们已经收集在

In [406]: selected_pixels
Out[406]: 
[(1, 1),
 (1, 2),
 (2, 1),
 (1, 0),
 (0, 2),
 (2, 0),
 (2, 3),
 (2, 2),
 (0, 0),
 (1, 3),
 (0, 1),
 (0, 3)]

型
你应该能够计算出整数除法和模数如何将p转换为x和y。你不需要在循环中运行它。选择任何p和任何shape。

展开查看全部

赞(0）回复(0）举报 2024-01-08

我来回答

在2D numpy数组中随机选择索引，了解ChatGPT解决方案

1条答案

相关问题

热门标签

最新问答