Numpy 数组及其运算 01

1.创建数组

我前面写的几篇关于numpy数组创建的文章
Numpy数据类型及数组创建01
Numpy数据类型及数组创建02

2.测试两个数组的对应元素是否足够接近

import numpy as np
x=np.array([1,2,3,4.001,5])
y=np.array([1,1.999,3,4.01,5.1])
print(np.allclose(x,y))
print(np.allclose(x,y,rtol=0.2)) # 设置相对误差参数 
print(np.allclose(x,y,atol=0.2)) # 设置绝对误差参数
print(np.isclose(x,y)) # isclose测试每一对元素
print(np.isclose(x,y,atol=0.2))

False
True
True
[ True False  True False False]
[ True  True  True  True  True]

3.修改数组中的元素值

import numpy as np
x=np.arange(8)
print(x)
print(np.append(x,8)) # 在尾部追加一个元素
print(np.append(x,[9,10])) # 在尾部追加多个元素
print(np.insert(x,1,8)) # 插入一个元素
print(x) # 不影响原来的数组

x[0]=8 # 使用下标的形式原地修改元素值
print(x) # 原来的数组被修改了

y=np.array([[1,2,3],[4,5,6]])
y[0,2]=4 # 修改第1行第3列的元素
print(y)

y=np.array([[1,2,3],[4,5,6],[7,8,9]])
y[1:,1:]=1
print(y) # 把行和列的下标大于等于1的元素都改为1

# 同时修改多个值
y[1:,1:]=[1,2]
print(y)

y[1:,1:]=[[1,2],[3,4]]
print(y)

[0 1 2 3 4 5 6 7]
[0 1 2 3 4 5 6 7 8]
[ 0  1  2  3  4  5  6  7  9 10]
[0 8 1 2 3 4 5 6 7]
[0 1 2 3 4 5 6 7]
[8 1 2 3 4 5 6 7]
[[1 2 4]
 [4 5 6]]
[[1 2 3]
 [4 1 1]
 [7 1 1]]
[[1 2 3]
 [4 1 2]
 [7 1 2]]
[[1 2 3]
 [4 1 2]
 [7 3 4]]

4.数组与标量的运算

import numpy as np
x=np.array([1,2,3,4,5])

print(x*2)
print(x/2)
print(x//2)
print(x**3)
print(x+2)
print(x%3)
print(2**x)
print(2/x)
print(63//x)

[ 2  4  6  8 10]
[0.5 1.  1.5 2.  2.5]
[0 1 1 2 2]
[  1   8  27  64 125]
[3 4 5 6 7]
[1 2 0 1 2]
[ 2  4  8 16 32]
[2.         1.         0.66666667 0.5        0.4       ]
[63 31 21 15 12]

5.数组与数组的运算

import numpy as np
print(np.array([1,2,3,4])+np.array([4,3,2,1]))
print(np.array([1,2,3,4])+np.array([4]))

a=np.array([1,2,3])

print(a+a)
print(a-a)
print(a*a)
print(a/a)
print(a**a)

b=np.array(([1,2,2],[3,4,4],[5,6,6]))
print(a*b) # a的每个元素乘b对应的列

print(a+b) # a的每个元素加b对应的列

[5 5 5 5]
[5 6 7 8]
[2 4 6]
[0 0 0]
[1 4 9]
[1. 1. 1.]
[ 1  4 27]
[[ 1  4  6]
 [ 3  8 12]
 [ 5 12 18]]
[[2 4 5]
 [4 6 7]
 [6 8 9]]

6.数组排序

import numpy as np
x=np.array([3,1,2])
print(np.argsort(x)) # 升序，但返回的不是元素值而是元素下标值
print('排序后:',end='')
for i in np.argsort(x):
    print(x[i],end=' ')
print()    
print('最大值',x[x.argmax()])
print('最小值',x[x.argmin()])
y=np.array([1,0,5,8,9,4])
y.sort() # 原地排序
print(y)

z=np.random.randint(1,10,(2,5))
print(z)
z.sort(axis=1) # 横向排序
print(z)

[1 2 0]
排序后:1 2 3 
最大值 3
最小值 1
[0 1 4 5 8 9]
[[2 1 8 5 8]
 [1 7 3 6 1]]
[[1 2 5 8 8]
 [1 1 3 6 7]]

7.数组的内积运算

import numpy as np
x=np.array([1,2,3])
y=np.array([4,5,6])
# 三种方法，结果一样
print(np.dot(x,y))
print(x.dot(y))
print(sum(x*y))

32
32
32