我有一个csv文件,其中总是有两个第一列,但不同文件的列数不同。csv可以如下所示:
Gondi,4012,227,233,157,158,149,158
Gondi,4013,227,231,156,159,145,153
Gondu,4014,228,233,157,158,145,153
Gondu,4015,227,231,156,159,149,158
目前,我正在使用NumPy
,加载此数据的代码是:
import numpy as np
def readfile(fname):
with open(fname) as f:
ncols = len(f.readline().split(','))
name = np.loadtxt(fname, delimiter=',', usecols=[0],dtype=str)
ind = np.loadtxt(fname, delimiter=',', usecols=[1],dtype=int)
data = np.loadtxt(fname, delimiter=',', usecols=range(2,ncols),dtype=int)
return data,name,ind
我能更有效地使用pandas
做同样的事情吗?你知道吗
我认为可以使用^{} 和^{} 来选择第一列、第二列和其他列:
相关问题 更多 >
编程相关推荐