for文でのnp.allの使い方

Question

Python (競技プログラミング) を勉強し始めて数か月(主に仕事後)しか経っていないため不適切な質問や非効率なコードが表示されているかもしれませんが、ご容赦ください。
完全に独学なため、常識的なことでも教えていただけると助かります。
こんな書き方がある、でも示していただけると嬉しいです。

以下に前提条件や試したことを書きます。

入力として
stockには、2種類の商品の在庫数
actには、stockに対する作業の種類 (subは在庫から商品を減らし、addは商品を増やす)
changeには、それぞれの作業におけるstockの変化量を示しており、それぞれリストに入っています。

出力として
在庫が減らす分、存在する場合は在庫数をカンマ区切りで出力し
どちらかの商品の在庫が足りない場合は"Not enough"を出力するとします。
("sub"の数だけ出力がある)

パターン1と2の場合は問題なく動きますが、np.allを用いたパターン3の場合は在庫数が負でも出力されます。

どうしてそうなるのかわからず、解決法を教えていただければ幸いです。

該当のソースコード

パターン１

import numpy as np
stock = [3,2]
act = ["sub", "sub", "add", "sub", "sub"]
change = [[1,2], [3,0], [3, 2], [2,1], [2,2]]
ans = []
for a, b in zip(act, change):
    tes = []
    for h, b2 in zip(stock, b):
        tes.append(h-b2)
    if a == "sub":
        if all((x>=0 for x in tes)):
            for e, (p, b2) in enumerate(zip(stock, b)):
                stock[e] -= b2 #stockの個数を変更
            ans.append(",".join([str(a) for a in stock]))
        else:
            ans.append("Not enough")
    elif a =="add":
        for e, (p,b2) in enumerate(zip(stock, b)):
            stock[e] += b2
for a in ans:
    print(a)

出力
2,0
Not enough
3,1
Not enough

パターン２

import numpy as np
stock = [3,2]
act = ["sub", "sub", "add", "sub", "sub"]
change = [[1,2], [3,0], [3, 2], [2,1], [2,2]]
ans = []
for a, b in zip(act, change):
    tes = (np.array(stock)-np.array(b)).tolist()
    if a == "sub":
        if all((x>=0 for x in tes)):
            for e, (p, b2) in enumerate(zip(stock, b)):
                stock[e] -= b2 #stockの個数を変更
            ans.append(",".join([str(a) for a in stock]))
        else:
            ans.append("Not enough")
    elif a =="add":
        for e, (p,b2) in enumerate(zip(stock, b)):
            stock[e] += b2
for a in ans:
    print(a)'

出力
2,0
Not enough
3,1
Not enough

パターン3

import numpy as np
stock = [3,2]
act = ["sub", "sub", "add", "sub", "sub"]
change = [[1,2], [3,0], [3, 2], [2,1], [2,2]]
ans = []
for a, b in zip(act, change):
    tes = np.array(stock)-np.array(b)
    if a == "sub":
        if np.all(x>=0 for x in tes): #これが上手く動作していない？
            for e, (p, b2) in enumerate(zip(stock, b)):
                stock[e] -= b2 #stockの個数を変更
            ans.append(",".join([str(a) for a in stock]))
        else:
            ans.append("Not enough")
    elif a =="add":
        for e, (p,b2) in enumerate(zip(stock, b)):
            stock[e] += b2
for a in ans:
    print(a)'

出力
2,0
-1,0
0,1
-2,-1

Comments

[y_waiwai]

デバッグぐらいしませんか

[hirochem]

アドバイスありがとうございます！
質問するときもデバックを入れたままの方がいいことを初めて知りました。次からはそうします。ありがとうございます！

Answer 1

tesがnumpyのndarrayなので、

if np.all(tes >= 0):

にすればよいです。tes >= 0がbooleanのndarrayになります。

パターン3は numpy.all(ジェネレータ) の書き方になっていますが、numpy.all()はジェネレータの中身を見ずに、ジェネレータそのもの(ジェネレータオブジェクト)として評価するようです。ジェネレータオブジェクトはFalseとして評価される値ではないので、numpy.all(ジェネレータ) は常にTrueを返すことになります。

Comments

[bsdfan]

あと、回答とは関係ないですが、パターン3で、numpyを使うなら、forを使わずに加減算するとか、もう少し工夫できると思います。

[hirochem]

返信いただきありがとうございます。
回答だけでなく、どうしてパターン３が上手くいかなかったか、まで解説していただいて非常に助かります。
そして、工夫のヒントをいただいたので実際にやってみたところ
stock = (np.array(stock) - np.array(b)).tolist()　をfor文のところと入れ替えることで上手くいきました！

numpyでリストの値を変化させてからリストに戻す、ということが身に付きました。ありがとうございます。

Answer 2

np.all(x>=0 for x in tes)をnp.all([x>=0 for x in tes])に修正しました。

import numpy as np
stock = [3,2]
act = ["sub", "sub", "add", "sub", "sub"]
change = [[1,2], [3,0], [3, 2], [2,1], [2,2]]
ans = []
for a, b in zip(act, change):
    tes = np.array(stock)-np.array(b)
    if a == "sub":
        if np.all([x>=0 for x in tes]):
            for e, (p, b2) in enumerate(zip(stock, b)):
                stock[e] -= b2 #stockの個数を変更
            ans.append(",".join([str(a) for a in stock]))
        else:
            ans.append("Not enough")
    elif a =="add":
        for e, (p,b2) in enumerate(zip(stock, b)):
            stock[e] += b2
for a in ans:
    print(a)

Comments

[hirochem]

ありがとうございます！
綺麗に動きました！
自分ではなかなか気づけないところだと思うので非常に助かります。