numpy loadtxt StopIteration error

trainfile = open('train.csv') 
features_train = np.loadtxt(trainfile, delimiter=',', usecols = (2,4,5,9), skiprows = 1) 
labels_train = np.loadtxt(trainfile, usecols = (1,), skiprows = 1)

它在第三行中抛出StopIteration错误。numpy loadtxt StopIteration error

来源

2017-09-09 Abdelrhman Mostafa

是的，因为你用尽了文件迭代器。 –

如果您不想引发异常，则可以使用'try/except'来优雅地处理此问题。 – JacobIRR

您应该加载一次和加载后的数据拆分：

trainfile = open('train.csv') 
all_data = np.loadtxt(trainfile, delimiter=',', usecols = (1,2,4,5,9), skiprows = 1) 
features_train = alldata[1:, ...] 
labels_train = alldata[0, ...]

来源

2017-09-09 18:45:24

numpy loadtxt StopIteration error

回答

相关问题