関数の引数で受け取るラムダ式を判別したい

Question

お世話になっております。
下記のような真偽値表が出力できるリストを作成したいです。
ラムダ式を引数でとるのですが、どう扱ったらよいのかわかりません。
ぜひ、ご回答お願いいたします。

#a,bがTrueならTrue。それ以外はFalse
def AND(a,b):
    return a and b

#aかbどちらか一方がTrueならTrue。それ以外はFalse
def OR(a,b):
    return a or b

#aとbがどちらも同じならTrue。それ以外はFalse
def EQ(a,b):
    return a == b


def table(a):


(a)には【lambda a, b: AND(a, OR(a, b))】か
       【lambda a, b, c: EQ(AND(a, OR(b, c)), OR(AND(a, b), AND(a, c)))】が入る想定。

<lambda a, b: AND(a, OR(a, b))入力時の想定>
[
 [True, True, True],
 [True, False, True],
 [False, True, False],
 [False, False, False]
]

<lambda a, b, c: EQ(AND(a, OR(b, c)), OR(AND(a, b), AND(a, c)))入力時の想定>
[
 [True, True, True, True],
 [True, True, False, True],
 [True, False, True, True],
 [True, False, False, True],
 [False, True, True, True],
 [False, True, False, True],
 [False, False, True, True],
 [False, False, False, True]
]

試したこと

from itertools import product

def table(a): 
    lamb = a
    ls = []

下記でそれぞれ出力したい結果は得られるのですが、どうにか条件等指定して、
欲しい結果だけを出力させたいです。

    for x,y in product([True,False],repeat=2):
            ls.append([x,y,lamb(x,y)])

    for x,y,z in product([True,False],repeat=3):
            ls.append([x,y,z,lamb(x,y,z)])

    return ls

Answer 1

~~ラムダ式の引数の数も渡してあげるようにすると、~~以下のように書けます。

from itertools import product
from pprint import pprint

#a,bがTrueならTrue。それ以外はFalse
def AND(a,b):
    return a and b

#aかbどちらか一方がTrueならTrue。それ以外はFalse
def OR(a,b):
    return a or b

#aとbがどちらも同じならTrue。それ以外はFalse
def EQ(a,b):
    return a == b

def table(a):
    n = a.__code__.co_argcount # 引数の数
    ls = []
    for vals in product([True,False], repeat=n):
        ls.append(list(vals) + [a(*vals)])

    return ls;

ls = table( lambda a, b: AND(a, OR(a, b)))
pprint(ls)
"""
[[True, True, True],
 [True, False, True],
 [False, True, False],
 [False, False, False]]
"""

ls = table( lambda a, b, c: EQ(AND(a, OR(b, c)), OR(AND(a, b), AND(a, c))))
pprint(ls)
"""
[[True, True, True, True],
 [True, True, False, True],
 [True, False, True, True],
 [True, False, False, True],
 [False, True, True, True],
 [False, True, False, True],
 [False, False, True, True],
 [False, False, False, True]]
"""

Comments

[Sayre]

回答ありがとうございます！
引き数をとるもので「__code__.co_argcount」というのがあるんですね。
少々質問させていただきたいのですが、
これはラムダ式の引数を取る以外にも使えるのでしょうか？

また、 [a(*vals)]の部分で、もしrepeat = 2の場合、valsに渡される値は
for vals in product([True,False],repeat=2):
print(*vals)
True True
True False
False True
False False
の形になりますが、a(*vals)とすることで、
a(True,True)の形になるということでしょうか？
そのまま代入されるとa(True True)のようになってしまうのではと思いましたが、ちゃんとラムダ式に代入されているので疑問に思いました。

[can110]

> これはラムダ式の引数を取る以外にも使えるのでしょうか？
はい。普通の関数でも使えます。

「a(*vals)」はPythonのアンパックという機能を利用しています。
「python 引数 アンパック」で検索すると解説ページがいろいろ見つかりますので参照ください。

[Sayre]

回答ありがとうございます。とても勉強になりました！