From an array with three columns, I want to be able to just take a slice of data from all three columns where the values in the first column are equal to the values defined in above.
above = {1, 5, 10}
data = np.arange(9).reshape(-1, 3)
energies = np.hsplit(data, 3)[0]
slice = set(energies) & above
The above comes back with:
Traceback (most recent call last):
File "<pyshell#27>", line 1, in <module>
slice = set(energies) & above
TypeError: unhashable type: 'numpy.ndarray'
How do I resolve this error?