您的变量
energies的形状可能不正确:
>>> from numpy import array>>> set([1,2,3]) & set(range(2, 10))set([2, 3])>>> set(array([1,2,3])) & set(range(2,10))set([2, 3])>>> set(array([[1,2,3],])) & set(range(2,10))Traceback (most recent call last): File "<stdin>", line 1, in <module>TypeError: unhashable type: 'numpy.ndarray'
如果使用这种方法读取列数据,就会发生以下情况:
>>> dataarray([[ 1., 2., 3.], [ 3., 4., 5.], [ 5., 6., 7.], [ 8., 9., 10.]])>>> hsplit(data,3)[0]array([[ 1.], [ 3.], [ 5.], [ 8.]])
也许你可以简单地使用
>>> data[:,0]array([ 1., 3., 5., 8.])
代替。
(PS您的代码看起来像它的犹豫不决它是否
data还是
elementdata。我认为它只是一个错字。)



