Извлечь ненулевые элементы каждой строки массива
Я использую Python 3.9/NumPy 1.22.
Предположим, у меня есть матрица 3х3:
import numpy as np
x = np.array([[10, 40, 0],
[0, 40, 90],
[10, 0, 90]])
Все элементы являются целыми числами >= 0. В каждой строке содержится ровно два ненулевых целых числа.
Я хотел бы извлечь ненулевые целые числа, чтобы создать матрицу 3x2 y такую, что
y = np.array([[10, 40],
[40, 90],
[10, 90]])
Я чувствую близость, когда использую numpy.apply_along_axis, numpy.squeeze и/или numpy.where, но мне чего-то не хватает.
«Все элементы являются целыми числами > 0» — явно нет. Некоторые - 0.
Ответы 2
Поскольку вы знаете, что у вас одинаковое количество нулей в строке, вы можете смело отбросить их и изменить форму:
x = np.array([[10,40,0],[0,40,90],[10,0,90]])
out = x[x!=0].reshape(len(x), -1)
Выход:
array([[10, 40],
[40, 90],
[10, 90]])
Ради интереса: если у вас разное количество нулей, вы можете переместить их в конец, а затем разрезать, чтобы их количество оставалось минимальным:
x = np.array([[10,40,0],[0,40,90],[0,0,90]])
# array([[10, 40, 0],
# [ 0, 40, 90],
# [ 0, 0, 90]])
m = x!=0
out = np.take_along_axis(x, np.argsort(~m, axis=1),
axis=1)[:, :m.sum(axis=1).max()]
# array([[10, 40],
# [40, 90],
# [90, 0]])
как обычно, я все усложняю... спасибо!
Вот одно решение, но оно требует знания точного количества ненулевых элементов (в данном случае 2)
x = np.array([[10,40,0],[0,40,90],[10,0,90]]) # exactly 2 non-zero per row
idx = np.where(x>0)
idx
Out[90]:
(array([0, 0, 1, 1, 2, 2], dtype=int64),
array([0, 1, 1, 2, 0, 2], dtype=int64))
y = x[idx].reshape((3,2))
y
Out[92]:
array([[10, 40],
[40, 90],
[10, 90]])
На самом деле, вы можете заменить 2 на -1, чтобы numpy вывел значение (что я и сделал в своем ответе);)
Если вы хотите использовать apply_along-axis, то:
y = np.apply_along_axis(lambda a: [b for b in a if b], 1, x)