从一个包含三列数据的文本文件中,我想要只获取第一列的值与
above
相等的行,并且同时获取这些行的三列数据。然后我想将获取到的数据存入名为slice
的新数组中(我使用的是Python 2.7)。above = range(18000, 18060, 5)
data = np.loadtxt(open('data.txt'), delimiter=None)
energies = (np.hsplit(data, 3))[0]
slice = set(energies)&set(above)
上面的输出是:
Traceback (most recent call last):
File "<pyshell#27>", line 1, in <module>
set(energies)&set(above)
TypeError: unhashable type: 'numpy.ndarray