def count(matrix: list, number: int):
count([[1, 4, 0, 0, 6, 3], [0, 3, 4, 0, 0, 0], [0, 0, 5, 6, 0]], 0)
I am trying to make function that counts all zeros or any other number from matrix list but for now its 0.
-
Because your input list contains only lists none of each equals 0. You need to flatten it firstYevhen Kuzmovych– Yevhen Kuzmovych2021-09-03 13:59:57 +00:00Commented Sep 3, 2021 at 13:59
-
So what does your code look like right now?Shinra tensei– Shinra tensei2021-09-03 14:00:58 +00:00Commented Sep 3, 2021 at 14:00
-
Also, does this answer your question? python .count for multidimensional arrays (list of lists)Yevhen Kuzmovych– Yevhen Kuzmovych2021-09-03 14:01:27 +00:00Commented Sep 3, 2021 at 14:01
Add a comment
|
2 Answers
try this:
def count(matrix: list, number: int):
return sum(l.count(number) for l in matrix)
count([[1, 4, 0, 0, 6, 3], [0, 3, 4, 0, 0, 0], [0, 0, 5, 6, 0]], 0)
# 9
count([[1, 4, 0, 0, 6, 3], [0, 3, 4, 0, 0, 0], [0, 0, 5, 6, 0]], 6)
# 2
Comments
import itertools
def count(matrix: list, number: int):
return list(itertools.chain.from_iterable(matrix)).count(number)
count([[1, 4, 0, 0, 6, 3], [0, 3, 4, 0, 0, 0], [0, 0, 5, 6, 0]], 0)
9
