在python中更改数组的内容

Question

我有一个多维数组（我们称之为“数据”），我打印它产生这样的结果

[[[255, 255, 255]
  [255, 255, 255]
  [0, 0, 0]
  [255, 255, 255]]
 [[0, 0, 0]
  [255, 255, 255]
  [0, 0, 0]
  [0, 0, 0]]
  ... and so on
  [255, 255, 255]]]

我想像这样更改数据的内容

[[1,
  1,
  0,
  1]
 [0,
  1,
  0,
  0]
  ... and so on
  1]]

[255, 255, 255] 变为 1，并且 [0, 0, 0]变成0

我正在尝试使用numpy.where，但我很绝望 在python编程中如何做到这一点？

Answer 1

一种方法是检查一个值是否为255，并使用np.logical_and 减少布尔结果

np.logical_and.reduce(a==255, axis=2).view('i1') 

---

以下示例：

a = np.array([[[255, 255 ,255],
               [255, 255, 255],
               [0 ,0, 0],
               [255, 255 ,255]],
              [[0 ,0, 0],
               [255, 255 ,255],
               [0 ,0, 0],
               [0 ,0, 0]]])

np.logical_and.reduce(a==255, axis=2).view('i1') 

array([[1, 1, 0, 1],
       [0, 1, 0, 0]], dtype=int8)

Answer 2

我会这样做：

使用tuple(col) == (255, 255, 255) 很重要，因为具有多个元素的数组的真值是不明确的。

data = np.asarray(im) # (assuming this is how you got your data)

new_data = np.array(np.zeros((len(data),len(data[0]))))

for x,row in enumerate(data):
    for y,col in enumerate(row):
        if tuple(col) == (255, 255, 255):
            new_data[x][y] = 1

print(new_data)

或者在一行中：

new_data = [[1 if tuple(col) == (255, 255, 255) else 0 for col in row] for row in data]

第一个将产生一个 numpy 数组，第二个将产生一个列表。但是，您可以使用以下方法将其转换为 numpy 数组：

new_data = np.array([[1 if tuple(col) == (255, 255, 255) else 0 for col in row] for row in data])

Answer 3

试试这个：

data = [
    [
        [255, 255, 255],
        [255, 255, 255],
        [0, 0, 0],
        [255, 255, 255]
    ],
    [
        [0, 0, 0],
        [255, 255, 255],
        [0, 0, 0],
        [0, 0, 0],
    ],
    [
        [255, 255, 255],
        [255, 255, 255],
        [0, 0, 0],
        [0, 0, 0],
    ]
]

result = [[1 if y == [255, 255, 255] else 0 for y in x] for x in data]
print(result)

请记住，如果唯一可能的值是 [255, 255, 255] 或 [0, 0, 0]，它将起作用（这是我从问题中理解的）