python - 即使文件只有一维,如何使 np.loadtxt 返回多维数组?

标签 python arrays numpy multidimensional-array

我需要获取 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]])?

最佳答案

刚找到here .

你可以要求它至少有 2 个维度:

arr = np.loadtxt('test.txt', ndmin=2)

关于python - 即使文件只有一维,如何使 np.loadtxt 返回多维数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/27855953/

相关文章:

python - 是否有 pandas 函数可以比较每行中的不同列值?

python - Google App Engine 模板循环遍历两个变量

python - groupby、transform 和 NaN 的奇怪结果

python - Blitz 代码产生不同的输出

c - 将字符串填充到结构体指针数组中

python - 如何将 numpy 数组写入 csv 文件?

处理二进制文件的python性能

python - 从字典更新列中组的值

c - 在链表中存储结构数组

java - 访问私有(private)数组中对象的私有(private)变量