如果满足条件，则在 Numpy 中替换行

Question

我在 NumPy 中有一个矩阵：

[[10, 10],
 [11, 10],
 [12, 10],
 [13, 10],
 [14, 10]]

我的目标是在满足左侧元素的特定条件时替换一行。

例如：

我有价值观：z= 13, x=1, y=0。我想要类似的东西：

如果左侧元素之一等于 z，则将该行替换为 [x y]

到目前为止，我写的是这样的：

import numpy as np
values = np.array([[10, 10],
     [11, 10],
     [12, 10],
     [13, 10],
     [14, 10]])
z = 13
x = 1
y = 0
values = np.where(values == [z, ], np.array([x, y]), values)

代码可以很好地替换左侧元素（x 已更新），但 y 出于某种原因保持不变或以某种方式获取另一行的值。

有谁知道我应该怎么做，或者是否有其他方法？

Answer 1

尝试values[values[:, 0] == z] = [x, y] - 即，将第一个条目为z 的行设置为[x, y]。

import numpy as np

values = np.array([[10, 10],
     [11, 10],
     [12, 10],
     [13, 10],
     [14, 10]])
z = 13
x = 1
y = 0

values[values[:, 0] == z] = [x, y]

print(values)