получить результаты numy вложенных массивов в отдельных вложенных массивах при выполнении операции без использования цикла for

Question

Приведенный ниже код проверяет, больше ли каждый элемент вложенных массивов, чем 2, и выдает результат в соответствующих вложенных массивах:

2d_array=np.array([[1,2,3,4],[4,56,7,1]])
for elem in 2d_array:
    print(elem[elem[:]>2])

Выход:

[3 4]
[ 4 56  7]

Можем ли мы сделать то же самое без использования for loop, предпочтительно с использованием numpy функций.

Answer 1

Numpy производит массивы. В вашем случае каждая результирующая строка имеет разную длину, поэтому вы не можете получить массив, просто список списков.

Однако, если вы хотите собрать все значения в одномерный массив, вы можете просто сделать

vaulues = array_2d[array_2d > 2]