numpy genfromtxt读取csv的第一个值是否丢失?

时间:2019-05-09 18:52:47

标签: python numpy genfromtxt

我正在尝试使用numpy的genfromtxt将键长度和能量的csv读取到数组中(用于生成势能面和反应路径,因此我将使用scipy.interpolate-因此需要每个值...)。

问题在于genfromtxt正在将每个csv输入的第一个值读取为NaN。我该如何解决?

作为示例,我在dcm_oh_lengths.csv中具有以下数据:

1.0763,1.1263,1.1763,1.2263,1.2763,1.3263,1.3763,1.4263,1.4763,1.5263,1.5763

我称之为

oh_all = np.genfromtxt(solv+'_oh_lengths.csv',dtype=float,delimiter=',')

然后oh_all返回

array([   nan, 1.1263, 1.1763, 1.2263, 1.2763, 1.3263, 1.3763, 1.4263,
       1.4763, 1.5263, 1.5763])

因此第一个数据点被读取为丢失。如果我将数据更改为

,1.0763,1.1263,1.1763,1.2263,1.2763,1.3263,1.3763,1.4263,1.4763,1.5263,1.5763

然后做同样的事情就会返回

array([   nan, 1.0763, 1.1263, 1.1763, 1.2263, 1.2763, 1.3263, 1.3763,
       1.4263, 1.4763, 1.5263, 1.5763])

作为一个较长的示例,能量的前几行(dcm_energies.csv)为:

-7162979.201,-7163010.482,-7163033.634,-7163043.279,-7163060.113,-7163068.894,-7163076.255,-7163078.541,-7163080.908,-7163056.179,-7163081.743
-7163005.74,-7163031.808,-7163050.794,-7163056.603,-7163064.619,-7163070.65,-7163080.606,-7163080.682,-7163081.125,-7163052.444,-7163078.824
-7163024.746,-7163046.199,-7163061.278,-7163063.603,-7163068.336,-7163071.692,-7163079.11,-7163077.25,-7163075.861,-7163043.325,-7163070.561 (...)

并通过上面的genfromtxt调用它可以得到:

array([[         nan, -7163010.482, -7163033.634, -7163043.279,
        -7163060.113, -7163068.894, -7163076.255, -7163078.541,
        -7163080.908, -7163056.179, -7163081.743],
       [-7163005.74 , -7163031.808, -7163050.794, -7163056.603,
        -7163064.619, -7163070.65 , -7163080.606, -7163080.682,
        -7163081.125, -7163052.444, -7163078.824],
       [-7163024.746, -7163046.199, -7163061.278, -7163063.603,
        -7163068.336, -7163071.692, -7163079.11 , -7163077.25 ,
        -7163075.861, -7163043.325, -7163070.561], (...)

2 个答案:

答案 0 :(得分:0)

我的猜测是文件以"byte order mark" (BOM)开头。文件是如何创建的?

尝试一下:

with open('dcm_oh_lengths.csv', 'r', encoding='utf-8-sig') as f: 
    oh_all = np.genfromtxt(f, dtype=float, delimiter=',')

答案 1 :(得分:0)

正如 Warren 指出的,这是一个 BOM 问题。

我在网上找到的一个可能更简单的解决方案是在记事本++中打开您的 CSV 文件。 如果您有 UTF-8 BOM 文件,您可以在右下角看到。

如果这样做,您只需单击编码并选择 UTF-8,然后保存您的文件。 这种方式无需添加更多代码。