Python numpy，获取数组最大值的位置，创建全为0(或1)的数组，创建单位矩阵

 

demo.py（np.argmax，np.zeros，np.ones，np.eye）：

# coding=utf-8
import numpy as np


t1 = np.arange(12, 24).reshape((3,4))
print(t1)
'''
[[12 13 14 15]
 [16 17 18 19]
 [20 21 22 23]]
'''
# 获取数组最大值、最小值的位置
a = np.argmax(t1)
print(a)  # 11 （位置从0开始算）

b = np.argmin(t1, axis=0)  # axis指定坐标轴。每一列最小值的位置。
print(b)  # [0 0 0 0]


# 创建全为0的数组(3行4列)
t2 = np.zeros((3, 4))
print(t2)
'''
[[0. 0. 0. 0.]
 [0. 0. 0. 0.]
 [0. 0. 0. 0.]]
'''
# 创建全为1的数组(3行4列)
t3 = np.ones((3, 4))
print(t3)
'''
[[1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]]
'''


# 创建单位矩阵(对角线全为1的方阵)
t4 = np.eye(3)
print(t4)
'''
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]
'''