语法

numpy.delete(arr, obj, axis=None)

例子

删除一行

import numpy as np

dataset=[[1,2,3],[2,3,4],[4,5,6]]

# 删除第一行
dataset = np.delete(dataset, 0, axis=0)

print(dataset )

执行结果：

[[2 3 4]
 [4 5 6]]

删除多行

import numpy as np

dataset=[[1,2,3],[2,3,4],[4,5,6]]
# 删除第1行和第3行
dataset = np.delete(dataset, [0,2], axis=0)

print(dataset )

执行结果：

[[2 3 4]]

删除一列

import numpy as np

dataset=[[1,2,3],[2,3,4],[4,5,6]]
dataset = np.delete(dataset, -1, axis=1)

print(dataset )

执行结果：

[[1 2 3]
 [2 3 4]]

删除多列

import numpy as np

dataset=[[1,2,3],[2,3,4],[4,5,6]]

# 删除第1列和第2列
dataset = np.delete(dataset, [0,1], axis=1)

print(dataset )

执行结果：

[[3]
 [4]
 [6]]

原文出处：/show_1EF5aYx6y7Nw.html