使用 numpy 行选择器,np.r_
.
>>> np.r_[range(3), range(15, 18), range(100, 103)]
或者(使用 hpaulj 的建议),
>>> np.r_[0:3, 15:16, 100:103]
array([ 0, 1, 2, 15, 16, 17, 100, 101, 102])
对于您的代码,您可以这样称呼它 -
numpy.loadtxt(
input_file_name,
delimiter=",",
skiprows = 1,
usecols=np.r_[range(1, 99), range(101, 199), range(74999, 77999)]
)