我正在尝试加载存储在CSV
文件中的一些数据,其中标题位于第一列。我正在使用numpy.loadtxt
(v1.6)。
我想知道是否有办法加载除第一个以外的所有列?
我知道可以选择我们要加载的列。问题是,我不知道有多少列(我只知道有超过255列)。
如果某人有诀窍,那就太棒了!
答案 0 :(得分:6)
好吧,它看起来像you can feed it a generator,所以只需删除生成器中的第一列:
def strip_first_col(fname, delimiter=None):
with open(fname, 'r') as fin:
for line in fin:
try:
yield line.split(delimiter, 1)[1]
except IndexError:
continue
data = np.loadtxt(strip_first_col('myfilename'))