从包含三列数据的文本文件中,我希望能够从所有三列中获取slice
个数据,其中第一列中的值等于{{1}中定义的值}。然后我想把数据片段放到一个名为above
的新数组中(我正在使用 Python 2.7 )
slice
以上回复:
above = range(18000, 18060, 5)
data = np.loadtxt(open('data.txt'), delimiter=None)
energies = (np.hsplit(data, 3))[0]
slice = set(energies)&set(above)
答案 0 :(得分:30)
您的变量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'
如果你使用你的方法阅读柱状数据会发生什么:
>>> data
array([[ 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.])
代替。
(P.S。您的代码看起来尚未确定是data
还是elementdata
。我认为这只是一个错字。)
答案 1 :(得分:3)
numpy.ndarray
可以包含任何类型的元素,例如int
,float
,string
等。检查类型并在必要时进行转换。