如何使用Numpy正确读取多列数据文件的第一列?
解决方法
你的问题出在pd.read_csv的默认分隔符设置上——你的数据是多空格分隔,但read_csv默认用逗号分隔,所以每一行被当成了单个字符串存在一列里,自然切片后还是整行内容。下面提供两种可靠的解决方案:
方法1:用Pandas正确读取后转Numpy
先指定分隔符为任意数量的空格,再提取第一列转成Numpy数组:
import pandas as pd # 用sep='\s+'匹配任意数量的空格作为分隔符 df = pd.read_csv('/home/psdlab/test1.dat', sep='\s+', header=None) # 提取第一列并转成Numpy数组 x_data = df.iloc[:, 0].values print(x_data)
方法2:直接用Numpy读取
如果不需要Pandas,直接用numpy.loadtxt更简洁,它默认处理空格分隔的数据:
import numpy as np # 读取文件,提取第一列 x_data = np.loadtxt('/home/psdlab/test1.dat')[:, 0] print(x_data)
两种方法都会输出第一列的数值数组,示例输出:
[ 59.806 -49.707 54.281 ... -65.698 -45.539 34.794]
内容的提问来源于stack exchange,提问作者bt22d075 Shradha sharma
相关产品推荐
相关产品推荐

