2

我正在使用以下代码从文本文件中读取：

import numpy as np
my_data = np.genfromtxt(resultsDirectory+'/Points.txt', delimiter=' ')
PointX = my_data[:,5]
PointY = my_data[:,11]

我的输入文件通常是这样的 -

ParamA : 0 ParamB : 7 ParamC : 0 ParamD : 1 Result : FAIL Time : 0 Epsilon : 0.5
ParamA : 0 ParamB : 11 ParamC : 0 ParamD : 1 Result : FAIL Time : 2 Epsilon : 0.5
ParamA : 0 ParamB : 7 ParamC : 0 ParamD : 3 Result : FAIL Time : 2 Epsilon : 0.25
ParamA : 0 ParamB : 13 ParamC : 0 ParamD : 1 Result : FAIL Time : 7 Epsilon : 0.25
ParamA : 0 ParamB : 7 ParamC : 0 ParamD : 4 Result : FAIL Time : 8 Epsilon : 0.125
ParamA : 0 ParamB : 8 ParamC : 0 ParamD : 2 Result : FAIL Time : 1 Epsilon : 0.125
ParamA : 0 ParamB : 8 ParamC : 0 ParamD : 3 Result : FAIL Time : 3 Epsilon : 0.125
ParamA : 0 ParamB : 8 ParamC : 0 ParamD : 4 Result : FAIL Time : 6 Epsilon : 0.125
ParamA : 0 ParamB : 9 ParamC : 0 ParamD : 2 Result : FAIL Time : 6 Epsilon : 0.125
ParamA : 0 ParamB : 10 ParamC : 0 ParamD : 2 Result : FAIL Time : 5 Epsilon : 0.125
ParamA : 0 ParamB : 14 ParamC : 0 ParamD : 1 Result : FAIL Time : 6 Epsilon : 0.125

当我从中提取 PointX 时，我得到

PointX = [7 11 7 13 7 8 8 8 9 10 14]
PointY = [1 1 3 1 4 2 3 4 2 2 1]

现在，有时我的文本文件只包含一行，甚至是空的。

例如，如果文本文件只有一行，则 my_data 数组是这样的 -

[  nan   nan   0.    nan   nan   7.    nan   nan   0.    nan   nan   1.
   nan   nan   nan   nan   nan  0.    nan   nan   0.5]

在这种情况下，my_data.shape返回(21,)。

但是读取数组PointX或PointY给我一个错误IndexError: invalid index。我想要PointX=[7]和PointY=[1]。或者如果文本文件是空的，它应该是PointX=[]and PointY=[]。

我应该如何解决这个问题？我还需要 my PointXandPointY成为一个数组，以免破坏依赖它的代码。

谢谢你。

4

2 回答 2

3

不幸的是，genfromtxt如果给定的文件只有一行，则返回一维数组，如果给定多行，则返回二维数组。您可以通过重塑来处理差异：

import numpy as np
my_data = np.genfromtxt('data', delimiter=' ')
if my_data.ndim == 1:
    my_data = my_data.reshape(1, -1)
PointX = my_data[:,5]
PointY = my_data[:,11]
print(PointX)
print(PointY)

产量

[ 7.]
[ 1.]
于 2012-10-11T16:12:55.713 回答
2

处理空文件：

import numpy as np
try:
    my_data = np.genfromtxt('data_file', delimiter=' ')
    if my_data.ndim == 1:
        my_data = my_data.reshape(1, -1)
    PointX = my_data[:,5]
    PointY = my_data[:,11]
except IOError: # File is empty
    PointX = []
    PointY = []

print(PointX)
print(PointY)
于 2012-10-11T16:28:08.903 回答