2021/9/28 python学习笔记：np.argmax、np.delete

1、np.argmain,np.argmax函数：找最大最小值索引，可按轴查找

import numpy as np

c = np.array([[4,7,0,9],[3,6,2,4],[9,8,0,3]])

print(np.argmax(c,axis=0))  # 按照轴0查找时，数组有3个元素，对这3个元素找最大，因为每个元素有4个子元素，故重复4次，输出4次找到的索引

输出：[2 2 1 0]

2、np.delete函数：按索引删除数组中的元素

import numpy as np

c = np.array([[4,7,0,9],[3,6,2,4],[9,8,0,3]])

print(np.delete(c, 2, 1))  # 沿轴1删除数组c中的第3行元素

输出：[[4 7 9]
 [3 6 4]
 [9 8 3]]