Pythonリストに0が含まれる場合、0を無視して掛け算をする

Question

独学でPythonを学び始めた初心者です。
基本的なことで誠に恐縮ですが、お知恵を拝借できますと幸いです。

ゴール：
リストの0以外のすべての数字の乗算を計算する。
もし0が含まれる場合は無視し、0以外の数字を乗ずる。

期待する計算結果：
my_prod_nz([]) returns 1
my_prod_nz([0]) returns 1
my_prod_nz([1,2]) returns 2
my_prod_nz([1,2,3]) returns 6
my_prod_nz([1,2,0,3]) returns 6

if関数を使えば計算可能かと思料しますが、式が分からず苦戦しております。
恐れ入りますがご教示いただけますと幸いです。

Comments

[Zuishin]

0 を取り除いたリストに変換すればいいと思います。

[quickquip]

0が入ってない前提ならコードがかけるんでしょうか? "ここまでならできる"コードを書く方が回答しやすいですよ

[yncam]

ご助言ありがとうございます。また、失礼いたしました。正しくありませんが、以下のコードを書いておりました。removeの部分が正しくないというエラーがたっておりますので、今一度改善策を考えます。

def my_prod_nz(ls):
s = 1
if 0 in ls:
my_prod_nz.remove(0)
for x in ls:
s = s * x
return s
else:
for x in ls:
s = s * x
return s

my_prod_nz([1,2,0,3])

Answer 1

numpyを使うと簡単

pip install numpy==1.19.3

import numpy as np

def my_prod_nz(lst):
    arr = np.array(lst)
    arr = arr[arr != 0] # 0を排除
    val = int(np.prod(arr)) # 総積を求める(空なら1)
    return val

my_prod_nz([0])
# 1
my_prod_nz([])
# 1
my_prod_nz([1, 0, 5, 6])
# 30

Comments

[yncam]

ご親切にアドバイスありがとうございます。恥ずかしながらnumpyを存じ上げませんでしたので精進いたします。

Answer 2

こんなのでどうでしょうか？
リストの0と''を1で置換して、順番に掛け算してるだけです。

a1 = []
a2 = [0]
a3 = [1,2]
a4 = [1,2,3]
a5 = [1,2,0,3]

def my_prod_nz(a):
    non_zero = [1 if i == 0 or i == '' else i for i in a]
    val = 1
    for i in non_zero:
        val *= i
    
    return val


print(my_prod_nz(a1))
print(my_prod_nz(a2))
print(my_prod_nz(a3))
print(my_prod_nz(a4))
print(my_prod_nz(a5))

Comments

[yncam]

ご丁寧にありがとうございます。いただいた回答を確認し、理解を深めてまいります。お手数おかけしました。

Answer 3

再帰という手法も覚えておいてください。

def my_prod_nz(x):
    if x == []:
        return 1
    elif x[-1] != 0:
        return my_prod_nz(x[:-1]) * x[-1]
    else:
        return my_prod_nz(x[:-1])

実行結果は以下です。

>>> print(my_prod_nz([]))
1
>>> print(my_prod_nz([0]))
1
>>> print(my_prod_nz([1,2]))
2
>>> print(my_prod_nz([1,2,3]))
6
>>> print(my_prod_nz([1,2,0,3]))
6

Answer 4

累積処理は reduce を使うと書きやすい。

from functools import reduce

a1 = []
a2 = [0]
a3 = [1,2]
a4 = [1,2,3]
a5 = [1,2,0,3]

def my_prod_nz(lst):
    return reduce(lambda x, y: (x if x else 1) * (y if y else 1), lst, 1)

print(my_prod_nz(a1))  # 1
print(my_prod_nz(a2))  # 1
print(my_prod_nz(a3))  # 2
print(my_prod_nz(a4))  # 6
print(my_prod_nz(a5))  # 6

Answer 5

参考情報

• Multiply all numbers in the list (4 different ways)

https://makecode.microbit.org/_FarCXKhYYe37

まず最初に list の要素をすべて掛け算する方法を研究してください。

次に 0 を掛け算しない方法を考えてみてください。
list 中の 0 を １ にする
list 中の 0 を取り除いてから掛け算する
掛け算の処理の中で、 ０ならスキップする という条件分岐を組み込む
、、、

Answer 6

質問者のプログラムを添削すると、removeする対象は関数my_prod_nzじゃ無くて引き数lsなので、

def my_prod_nz(ls):
	s = 1
	if 0 in ls:
		ls.remove(0)
		for x in ls:
			s = s * x
		return s
	else:
		for x in ls:
			s = s * x
		return s

引き数オブジェクトを破壊してしまって良いのかという問題はありますが。

ただし、removeは最初の0しか削除しないので、0が複数あると駄目です。
0がある限り削除し続けると言うことで、

def my_prod_nz(ls):
	s = 1
	while 0 in ls:
		ls.remove(0)
	for x in ls:
		s = s * x
	return s

でしょうか。

Answer 7

謎のクラスを作ってみる。

class I(int):
   def __mul__(self, other):
     x = 1 if self == 0 else self.__int__()
     y = 1 if other == 0 else other.__int__()
     return I(x * y)

>>> I(1) * I(2)
2
>>> I(1) * I(0) * I(3)
3
>>> x = [I(n if n % 2 == 0 else 0) for n in range (10)]
>>> x
[0, 0, 2, 0, 4, 0, 6, 0, 8, 0]
>>> import functools
>>> functools.reduce(lambda x, y: x * y, x)
384