Как убрать нулевые строки и колонки из двух-мерной матрицы?

Question

[pcdesign]

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

Чтобы получилось:

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

Answer 1

[Andrey Dugin]

list(zip(*filter(any, zip(*filter(any, matrix)))))

UPD Вынесу из комментов окончательный вариант:

def trim(matrix):
    fltr = lambda matrix: zip(*filter(any, matrix))
    return [list(row) for row in fltr(fltr(matrix))]

matrix = trim(matrix)

Comments

[Сергей Соколов]

Магия! (первый однострочник)
Но возвращает туплы: `[(4, 0, 1), (0, 2, 3), (0, 3, 1)]` Как сделать, чтобы вернула список списков?

[Andrey Dugin]

Сергей Соколов, zip возвращает список туплов, нужно преобразовать:

def trim(matrix):
    fltr = lambda m: zip(*filter(any, m))
    return [list(row) for row in fltr(fltr(matrix))]

matrix = trim(matrix)

[Сергей Соколов]

Andrey Dugin, спасибо! С (монти) Пайтоном мне ещё разбираться и разбираться. На PHP/JS которые много лет использую в работе, такие фишки легко получаются, а с Python чувствую себя второгодником в начальной школе )

[Andrey Dugin]

Сергей Соколов, пожалуйста. Тщательно изучите встроенные функции.

Answer 2

[AlexandrBirukov]

без сторонних либ

start = [
    [0, 4, 0, 1, 0],
    [0, 0, 2, 3, 0],
    [0, 0, 3, 1, 0],
    [0, 0, 0, 0, 0],
]
result = [list(t) for t in zip(*[i for i in zip(*[x for x in start if any(x)]) if any(i)])]

Comments

[Andrey Dugin]

Слишком много букв)
list(zip(*filter(any, zip(*filter(any, matrix)))))

[AlexandrBirukov]

Andrey Dugin, нет предела совершенству, но я думал прежде всего о производительности, вроде как генератор списков быстрее работает чем функции

[Andrey Dugin]

AlexandrBirukov, сомнительный прирост скорости в ущерб читаемости. Золотое правило - не надо ничего оптимизировать, не проведя профилирование программы. Бутылочное горлышко может оказаться вовсе не здесь. Да и требования по скорости могут быть не актуальны.

[AlexandrBirukov]

Andrey Dugin, полностью согласен, ваше решение выглядит изящнее как минимум, ну я как смог, я точно не гений питона

Answer 3

[Сергей Соколов]

С Python только разбираюсь, ответ не претендует на оптимальность.

import numpy as np

def noZeros(a):
    arr = np.array(a)
    rows = arr[ ~np.all(arr == 0, axis = 1)]
    colsTrue = ~np.all(rows == 0, axis = 0)
    return rows[:, colsTrue == True]

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

noZeros(a)

Результат:

array([[4, 0, 1],
       [0, 2, 3],
       [0, 3, 1]])

Comments

[pcdesign]

Спасибо за ответ, но если в матрице есть строковоые значение, то ломается.

>>> import numpy as np
>>> 
>>> 
>>> def noZeros(a):
...     arr = np.array(a)
...     rows = arr[ ~np.all(arr == 0, axis = 1)]
...     colsTrue = ~np.all(rows == 0, axis = 0)
...     return rows[:, colsTrue == True]
... 
>>> a = [
...        [0,'a',0,1,0],
...        [0,0,'s',3,0],
...        [0,0,3,1,0],
...        [0,0,0,0,0],
... ]
>>> noZeros(a)
__main__:3: FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 3, in noZeros
  File "/home/w2/env/lib/python3.4/site-packages/numpy/core/fromnumeric.py", line 2050, in all
    return arr.all(axis=axis, out=out, **kwargs)
  File "/home/w2/env/lib/python3.4/site-packages/numpy/core/_methods.py", line 41, in _all
    return umr_all(a, axis, dtype, out, keepdims)
numpy.core._internal.AxisError: axis 1 is out of bounds for array of dimension 0

[Сергей Соколов]

pcdesign, разумеется, ведь используется numpy – библиотека для работы с численными данными.