numpy 常用 api（四）

np.delete()

from sklearn.datasets import load_iris
import numpy as np

test_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]