Давайте получим A, как показано ниже:
A = np.array([[0, 0, 0, 0, 0],
[0, 1, 0, 0, 0],
[1, 0, 0, 0, 0],
[1, 0, 1, 0, 0],
[0, 0, 0, 0, 0]])
И теперь я должен взять все строки, которые имеют 1 в первом столбце:
A[A[:, 0] > 0]
return: array([[1, 0, 0, 0, 0],
[1, 0, 1, 0, 0]])
А теперь я хочуизменить первый столбец второй строки?Как мне это сделать ?Потому что это не работает:
A[A[:, 0] > 0][1, 1] = 1
A
array([[0, 0, 0, 0, 0],
[0, 1, 0, 0, 0],
[1, 3, 0, 0, 0],
[1, 0, 1, 0, 0],
[0, 0, 0, 0, 0]])
Мой код сложнее, чем этот, и поэтому условие должно быть выполнено в два раза, как здесь: (1)> 0 и (2) [1, 1].