根据另一列 np 设置列

Question

我想确保第 2 列小于第 1 列并且它刚设置为 0

x = np.array([[0,1],[1,0]])
x = np.where((x[1] > (x[0])), 0, x)

print(x)=>[[0,0],[1,0]]

Answer 1

也许这对你有帮助：

arr = np.array([[0,1],[1,0]])
arr[arr[:,1] > arr[:,0], 1] = 0
print(arr)

输出：

array([[0, 0],
       [1, 0]])

Answer 2

你从一个列表开始，所以我会给你一个列表答案。

首先定义一个简单的辅助函数：

def foo(row):
    if row[1]<row[0]:
        row[1] = 0
    return row

并将其逐行应用于 x：

In [37]: x = [[0,1],[1,0]]
In [38]: [foo(row) for row in x]
Out[38]: [[0, 1], [1, 0]]

Set column based on another column np