关于numpy.delete()
删除一列
dataset=[[1,2,3],[2,3,4],[4,5,6]]
import numpy as np
dataset = np.delete(dataset, -1, axis=1)
dataset结果为:
array([[1, 2],
[2, 3],
[4, 5]])
若要按行删除,将axis=0
dataset = np.delete(dataset, -1, axis=0)
输出结果为:
array([[1, 2, 3],
[2, 3, 4]])
删除多列
arr = np.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]])
np.delete(arr, [1,2], axis=1)
输出:
array([[ 1, 4],
[ 5, 8],
[ 9, 12]])