Numpy loadtxt跳过第一列

时间:2013-12-17 01:12:45

标签: python numpy

我正在尝试加载存储在CSV文件中的一些数据,其中标题位于第一列。我正在使用numpy.loadtxt(v1.6)。

我想知道是否有办法加载除第一个以外的所有列?

我知道可以选择我们要加载的列。问题是,我不知道有多少列(我只知道有超过255列)。

如果某人有诀窍,那就太棒了!

1 个答案:

答案 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'))