博客

1. numpy 矩阵
   
   【问题描述】
   
   在数组主对角线上创建一个值为1,2,3,4的4*4矩阵，图例如下：
   
   [[1. 0. 0. 0.]
   
   [0. 2. 0. 0.]
   
   [0. 0. 3. 0.]
   
   [0. 0. 0. 4.]]
   
   【输入形式】
   
   4
   
   【输出形式】
   
   [[1. 0. 0. 0.]
   
   [0. 2. 0. 0.]
   
   [0. 0. 3. 0.]
   
   [0. 0. 0. 4.]]

import numpy as np
n=int(input())
array=np.zeros([4,4],dtype=np.float)
for i in range(n):
    array[i][i]=i+1
print(array)

2. 创建ndarray
   
   【问题描述】
   
   创建一个长度为n的一维全为0的ndarray对象，然后让第m个元素等于1
   
   输入n,m并用空格隔开, 且n>=m
   
   【样例输入】10 5
   
   【样例输出】
   
   [[0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]]

import numpy as np
n,m=map(int,input().split())

array=np.zeros([1,n],dtype=np.float)
array[0][m-1]=1
print(array)

3.numpy矩阵形状

import numpy as np
def main():
	a=[1,2,3,4,5,6,7,8,9,10,11,12]
    n,m=input().split()
    n=int(n)
    m=int(m)
    a=np.array(a)
    if n*m!=12:
        print("NO")
    else:
        a.shape=n,m
        print(a)
if  __name__  ==  '__main__':
        main()

4.numpy数据排序

import numpy as np
def main():
	arr=input("")
    a=[float(x)  for  x  in  arr.split()]
    a=np.array(a)
    a=np.sort(a)
    print(a)
if  __name__  ==  '__main__':
        main()