我需要获取 ndarray
的最后四列数据,大多数时候代码 arr[:, -4:]
是可以的,但是如果数组只是有一个维度,这将抛出 IndexError: too many indices
。
我的数据是通过 arr = np.loadtxt('test.txt')
获取的,所以如果 test.txt
有不止一行,比如
0 1 2 3 4
0 10 20 30 40
一切正常,但是如果 test.txt
只有一行,比如
0 1 2 3 4
这将返回 array([ 0, 1, 2, 3, 4])
,然后 arr[:, -4:]
将抛出异常,因为它应该是arr[-4:]
,那么如何让loadtxt
返回array([[ 0, 1, 2, 3, 4]])
?
最佳答案
关于python - 即使文件只有一维,如何使 np.loadtxt 返回多维数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/27855953/