numpy 常用 api(四)

来源:互联网 发布:建津预算软件 编辑:程序博客网 时间:2024/06/14 06:21

np.delete()

属非更易型操作,是为了获取返回值,对原始数据不会进行修改;

from sklearn.datasets import load_irisimport numpy as nptest_idx = [0, 50, 150]X_train, y_train = np.delete(iris.data, test_idx, axis=0), np.delete(iris.target, test_idx)X_test, y_test = iris.data[test_idx], iris.target[test_idx]
0 0
原创粉丝点击