How can I generate a 2D boolean array using a list of tuples that shows the indices of the True values?
For example I have the following list of tuples:
Here's an approach using NumPy's linear indexing that works for tuples of any lengths intended to generate multi-dimensional arrays -
# Convert list of indices to a 2D array version
idx = np.array(lst)
# Decide on the shape of output array based on the extents, then initialize
shp = idx.max(0)+1
out = np.zeros(shp,dtype=bool)
# Using np.put insert 1s in out at places specified by linear indices version
np.put(out,np.ravel_multi_index(idx.T,shp),1)
Sample input, output -
In [54]: lst
Out[54]: [(0, 1, 3), (0, 2, 2), (1, 0, 0), (1, 3, 1), (2, 1, 3)]
In [55]: out
Out[55]:
array([[[False, False, False, False],
[False, False, False, True],
[False, False, True, False],
[False, False, False, False]],
[[ True, False, False, False],
[False, False, False, False],
[False, False, False, False],
[False, True, False, False]],
[[False, False, False, False],
[False, False, False, True],
[False, False, False, False],
[False, False, False, False]]], dtype=bool)