numpy广播(Broadcast)
若数组a,b形状相同,即a.shape==b.shape,那么a+b,a*b的结果就是对应数位的运算
import numpy as np
a=np.array([[1,2,3],[4,5,6]])
b=np.array([[1,4,7],[2,5,8]])
print(a+b,'
')
print(a*b)
[[ 2 6 10]
[ 6 10 14]]
[[ 1 8 21]
[ 8 25 48]]
若两个数组形状不同,且有一个数组维度为1,则会触发广播机制
a=np.array([[1,2,3],[4,5,6]])
b=np.array([1,2,3])
# 等同于np.array([[1,2,3],[4,5,6]])与np.array([1,2,3],[1,2,3])运算
print(a+b,'
')
print(a*b)
[[2 4 6]
[5 7 9]]
[[ 1 4 9]
[ 4 10 18]]
numpy数组操作函数
修改数组形状
numpy.reshape()
不改变数据的情况下修改形状
numpy.reshape(array , newshape , order = 'C')
参数 | 描述 |
---|---|
array | 要修改形状的数组 |
newshape | 整数或整数数组,新的形状应该兼容原有形状 |
order | 'C'——按行,'F'——按列,'A'——原顺序,'K'——元素咋内存中出现的顺序 |
import numpy as np
a_array=np.arange(16)
print(a_array,'
')
b_array=np.reshape(a_array,[4,4])
print(b_array,'
')
# 也可以打点调用
c_array=a_array.reshape([2,8])
print(c_array)
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[12 13 14 15]]
[[ 0 1 2 3 4 5 6 7]
[ 8 9 10 11 12 13 14 15]]
numpy.ndarray.flat
numpy.ndarray.flat为数组元素迭代器
array=np.arange(9).reshape([3,3])
print(array,'
')
# 按行遍历数组
for row in array:
print(row)
# 使用数组元素迭代器
for element in array.flat:
print(element)
[[0 1 2]
[3 4 5]
[6 7 8]]
[0 1 2]
[3 4 5]
[6 7 8]
0
1
2
3
4
5
6
7
8
numpy.ndarray.flatten和numpy.ravel
- numpy扁平化函数
numpy.ndarray.flatten
返回一份数组拷贝,对拷贝内容的修改不影响原始数值;numpy.ravel
返回一个数组的视图,修改视图时会影响原始数组
numpy.ndarray.flatten(order = 'C')
numpy.ravel(order = 'C')
参数 | 描述 |
---|---|
order | 'C'——按行,'F'——按列,'A'——原顺序,'K'——元素咋内存中出现的顺序 |
array=np.arange(16).reshape([4,4])
print(array,'
')
print(array.flatten(),'
')
print(array.ravel())
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[12 13 14 15]]
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]
a_array=np.arange(16).reshape([4,4])
print(a_array,'
')
#创建和a_array同样的数组b_array
b_array=a_array.copy()
c_array=a_array.ravel()
d_array=array.flatten()
print('c_array:')
print(c_array)
print('d_array:')
print(d_array,'
')
c_array[1]=100
d_array[1]=100
print('a_array:')
print(a_array)
print('b_array:')
print(b_array,'
')
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[12 13 14 15]]
c_array:
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]
d_array:
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]
a_array:
[[ 0 100 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[ 12 13 14 15]]
b_array:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[12 13 14 15]]
翻转数组
transpose和ndarray.T
- numpy数组转置函数
a_array=np.arange(16).reshape([4,4])
print('a_array:
',a_array)
print('使用transpose后:')
print(np.transpose(a_array))
print('使用.T转置后:')
print(a_array.T)
a_array:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[12 13 14 15]]
使用transpose后:
[[ 0 4 8 12]
[ 1 5 9 13]
[ 2 6 10 14]
[ 3 7 11 15]]
使用.T转置后:
[[ 0 4 8 12]
[ 1 5 9 13]
[ 2 6 10 14]
[ 3 7 11 15]]
numpy.swapaxes
- numpy用于交换数组两个轴的函数
numpy.swapaxes(arr , axis1, axis2)
参数 | 描述 |
---|---|
arr | 输入数组 |
axis1 | 对应数组第一个轴 |
axis2 | 对应数组第二个轴 |
array=np.arange(8).reshape(2,2,2)
print(array)
# 交换第零个轴和第二个轴
print(np.swapaxes(array,0,2))
[[[0 1]
[2 3]]
[[4 5]
[6 7]]]
[[[0 4]
[2 6]]
[[1 5]
[3 7]]]
(0)000->(0)000 | (1)001->(4)100 |
(2)010->(2)010 | (3)011->(6)110 |
(4)100->(1)001 | (5)101->(5)101 |
(6)110->(3)011 | (7)111->(7)111 |