TypeError：创建numpy数组时无法理解数据类型

Question

Having a list of boolean conditions list I would like to generate a matrix with the lists that have False or True values

For this example values = [[True, False], [False], [True], [True, False]] the result will be

    False  True
0   1       1
1   1       0
2   0       1
3   1       1

I tried to do that as following:

nodes = [True, False]
values = [[True, False], [False], [True], [True, False]]
res = np.array([[int(cond in vals) for vals in values] for cond in nodes],
                dtype=[(node, int) for node in nodes])

But I am getting the error TypeError: data type not understood

Answer 1

Try dtype=int , and then we have,

import numpy as np 

nodes = [True, False]
values = [[True, False], [False], [True], [True, False]]
res = np.array([[cond in vals for vals in values] for cond in nodes], dtype=int)

print(res)
# Output
[[1 0 1 1]
 [1 1 0 1]]