python - 在Numpy数组的列内进行条件替换

我有一个numpy数组，例如：

theData= [[0, 1, 1, 1],[0, 1, 3, 1],[3, 4, 1, 3],[0, 1, 2, 0],[2, 1, 0, 0]]

如何将第一列中的所有零替换为-1？
很容易用theData[theData==0] = -1替换整个数组中的所有零，所以我认为这样做是可行的

theData[theData[:,0] == 0] = -1
theData[:,0 == 0] = -1

但是对于第一列值为零的任何行，这些值都会将行中的所有值更改为-1。不是我的目标，我想把替换限制在第一列（或其他列）。
这显然可以通过循环来完成。也可以通过将第一列提取为1D数组，在其中进行替换，然后将其转置复制到原始的第一列上来完成。但我怀疑有一种更快、更像蟒蛇的方法可以做到这一点。也许用np.where，但我搞不懂。

最佳答案

只要不使用该列构建其他对象，就可以直接为该列编制索引。检查以下示例：

theData= np.array([[0, 1, 1, 1],[0, 1, 3, 1],[3, 4, 1, 3],[0, 1, 2, 0],[2, 1, 0, 0]])
print(theData)
theData[:,0][theData[:,0] == 0] = -1
print(theData)

结果是：

[[0 1 1 1]
 [0 1 3 1]
 [3 4 1 3]
 [0 1 2 0]
 [2 1 0 0]]
[[-1  1  1  1]
 [-1  1  3  1]
 [ 3  4  1  3]
 [-1  1  2  0]
 [ 2  1  0  0]]

关于python - 在Numpy数组的列内进行条件替换，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/49751455/